performance improvement

fieg · Aug 25, 2014 · e9bbc12 · e9bbc12
1 parent a99d11a
commit e9bbc12
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/src/Fieg/Bayes/Classifier.php b/src/Fieg/Bayes/Classifier.php
@@ -142,16 +142,15 @@ public function reset()
      */
     protected function inversedTokenCount($token, $label)
     {
-        $total = 0;
         $data = $this->data;
 
-        unset($data[$label]);
+        $totalTokenCount = $this->tokens[$token];
 
-        foreach ($data as $_label => $tokenCounts) {
-            $total += intval(@$tokenCounts[$token]);
-        }
+        $totalLabelTokenCount = intval(@$data[$label][$token]);
+
+        $retval = $totalTokenCount - $totalLabelTokenCount;
 
-        return $total;
+        return $retval;
     }
 
     /**