Changed SpamDetector to usage storage and more stop words

laravelsu · Mar 20, 2024 · 0ae7652 · 0ae7652
1 parent 2e21790
commit 0ae7652
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/app/Services/SpamDetector.php b/app/Services/SpamDetector.php
@@ -3,6 +3,7 @@
 namespace App\Services;
 
 use AssistedMindfulness\NaiveBayes\Classifier;
+use Illuminate\Support\Facades\Storage;
 use Illuminate\Support\Str;
 
 class SpamDetector
@@ -42,7 +43,8 @@ class SpamDetector
         'финансовая защита', 'инвестиции в будущее', 'реальный заработок',
         'финансовый инструментарий', 'инвестиции в золото', 'инвестиции в криптовалютные фонды',
         'увеличение прибыли в интернете', 'инвестирование в акции',
-        'финансовая безопасность',
+        'финансовая безопасность', 'нужен только телефон', 'стабильный доход',
+        'бесплатное обучение',
     ];
 
     /**
@@ -75,8 +77,8 @@ public function checkByClassifier(): bool
 
         // Train the classifier with spam and ham messages
         $this
-            ->trainClassifier($classifier, 'classifiers/spam.json', static::SPAM)
-            ->trainClassifier($classifier, 'classifiers/ham.json', static::HAM);
+            ->trainClassifier($classifier, 'spam.json', static::SPAM)
+            ->trainClassifier($classifier, 'ham.json', static::HAM);
 
         return $classifier->guess($this->message) === static::SPAM;
     }
@@ -92,7 +94,7 @@ public function checkByClassifier(): bool
      */
     private function trainClassifier(Classifier $classifier, string $fileName, string $label): self
     {
-        $messages = json_decode(file_get_contents(storage_path($fileName)));
+        $messages = json_decode(Storage::disk('classifiers')->get($fileName));
 
         foreach ($messages as $message) {
             $classifier->learn($message, $label);

diff --git a/config/filesystems.php b/config/filesystems.php
@@ -48,6 +48,12 @@
             'throw'  => false,
         ],
 
+        'classifiers' => [
+            'driver' => 'local',
+            'root'   => storage_path('classifiers'),
+            'throw'  => false,
+        ],
+
         'public' => [
             'driver'     => 'local',
             'root'       => storage_path('app/public'),