From 914b4da449a37a7529e153bc83b4b37d3df1eb92 Mon Sep 17 00:00:00 2001
From: fieg <jeroen@fieg.nl>
Date: Thu, 2 Jun 2016 13:19:37 +0200
Subject: [PATCH 1/2] removed error suppression

---
 src/Fieg/Bayes/Classifier.php | 29 +++++++++++++++++++++--------
 1 file changed, 21 insertions(+), 8 deletions(-)

diff --git a/src/Fieg/Bayes/Classifier.php b/src/Fieg/Bayes/Classifier.php
index 07a4492..24317e7 100644
--- a/src/Fieg/Bayes/Classifier.php
+++ b/src/Fieg/Bayes/Classifier.php
@@ -57,13 +57,26 @@ public function train($label, $text)
     {
         $tokens = $this->tokenizer->tokenize($text);
 
+        if (!isset($this->labels[$label])) {
+            $this->labels[$label] = 0;
+            $this->data[$label] = [];
+            $this->docs[$label] = 0;
+        }
+
         foreach ($tokens as $token) {
-            @$this->labels[$label]++;
-            @$this->tokens[$token]++;
-            @$this->data[$label][$token]++;
+            if (!isset($this->tokens[$token])) {
+                $this->tokens[$token] = 0;
+            }
+            if (!isset($this->data[$label][$token])) {
+                $this->data[$label][$token] = 0;
+            }
+
+            $this->labels[$label]++;
+            $this->tokens[$token]++;
+            $this->data[$label][$token]++;
         }
 
-        @$this->docs[$label]++;
+        $this->docs[$label]++;
     }
 
     /**
@@ -83,7 +96,7 @@ public function classify($text)
         foreach ($this->labels as $label => $labelCount) {
             $logSum = 0;
 
-            $docCount = intval(@$this->docs[$label]);
+            $docCount = intval($this->docs[$label]);
             $inversedDocCount = $totalDocCount - $docCount;
 
             if (0 === $inversedDocCount) {
@@ -91,13 +104,13 @@ public function classify($text)
             }
 
             foreach ($tokens as $token) {
-                $totalTokenCount = intval(@$this->tokens[$token]);
+                $totalTokenCount = isset($this->tokens[$token]) ? intval($this->tokens[$token]) : 0;
 
                 if (0 === $totalTokenCount) {
                     continue;
                 }
 
-                $tokenCount         = intval(@$this->data[$label][$token]);
+                $tokenCount         = isset($this->data[$label][$token]) ? intval($this->data[$label][$token]) : 0;
                 $inversedTokenCount = $this->inversedTokenCount($token, $label);
 
                 $tokenProbabilityPositive = $tokenCount / $docCount;
@@ -146,7 +159,7 @@ protected function inversedTokenCount($token, $label)
 
         $totalTokenCount = $this->tokens[$token];
 
-        $totalLabelTokenCount = intval(@$data[$label][$token]);
+        $totalLabelTokenCount = isset($data[$label][$token]) ? intval($data[$label][$token]) : 0;
 
         $retval = $totalTokenCount - $totalLabelTokenCount;
 

From ff287a493a613d7a3443bb67ee0eaea452f26e8f Mon Sep 17 00:00:00 2001
From: fieg <jeroen@fieg.nl>
Date: Thu, 2 Jun 2016 13:21:05 +0200
Subject: [PATCH 2/2] removed intval calls

---
 src/Fieg/Bayes/Classifier.php | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/Fieg/Bayes/Classifier.php b/src/Fieg/Bayes/Classifier.php
index 24317e7..8ca7eb2 100644
--- a/src/Fieg/Bayes/Classifier.php
+++ b/src/Fieg/Bayes/Classifier.php
@@ -96,7 +96,7 @@ public function classify($text)
         foreach ($this->labels as $label => $labelCount) {
             $logSum = 0;
 
-            $docCount = intval($this->docs[$label]);
+            $docCount = $this->docs[$label];
             $inversedDocCount = $totalDocCount - $docCount;
 
             if (0 === $inversedDocCount) {
@@ -104,13 +104,13 @@ public function classify($text)
             }
 
             foreach ($tokens as $token) {
-                $totalTokenCount = isset($this->tokens[$token]) ? intval($this->tokens[$token]) : 0;
+                $totalTokenCount = isset($this->tokens[$token]) ? $this->tokens[$token] : 0;
 
                 if (0 === $totalTokenCount) {
                     continue;
                 }
 
-                $tokenCount         = isset($this->data[$label][$token]) ? intval($this->data[$label][$token]) : 0;
+                $tokenCount         = isset($this->data[$label][$token]) ? $this->data[$label][$token] : 0;
                 $inversedTokenCount = $this->inversedTokenCount($token, $label);
 
                 $tokenProbabilityPositive = $tokenCount / $docCount;
@@ -159,7 +159,7 @@ protected function inversedTokenCount($token, $label)
 
         $totalTokenCount = $this->tokens[$token];
 
-        $totalLabelTokenCount = isset($data[$label][$token]) ? intval($data[$label][$token]) : 0;
+        $totalLabelTokenCount = isset($data[$label][$token]) ? $data[$label][$token] : 0;
 
         $retval = $totalTokenCount - $totalLabelTokenCount;