Changed SpamDetector

laravelsu · Mar 20, 2024 · 801d769 · 801d769
1 parent c5f9092
commit 801d769
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/app/Services/SpamDetector.php b/app/Services/SpamDetector.php
@@ -74,8 +74,9 @@ public function checkByClassifier(): bool
         $classifier = new Classifier();
 
         // Train the classifier with spam and ham messages
-        $this->trainClassifier($classifier, 'classifiers/spam.json', static::SPAM);
-        $this->trainClassifier($classifier, 'classifiers/ham.json', static::HAM);
+        $this
+            ->trainClassifier($classifier, 'classifiers/spam.json', static::SPAM)
+            ->trainClassifier($classifier, 'classifiers/ham.json', static::HAM);
 
         return $classifier->guess($this->message) === static::SPAM;
     }
@@ -87,14 +88,17 @@ public function checkByClassifier(): bool
      * @param string                                     $fileName   The path to the JSON file containing messages
      * @param string                                     $label      The label to assign to the messages (spam or ham)
      *
-     * @return void
+     * @return self
      */
-    private function trainClassifier(Classifier $classifier, string $fileName, string $label): void
+    private function trainClassifier(Classifier $classifier, string $fileName, string $label): self
     {
         $messages = json_decode(file_get_contents(storage_path($fileName)));
+
         foreach ($messages as $message) {
             $classifier->learn($message, $label);
         }
+
+        return $this;
     }
 
     /**