Merge pull request #20 from fieg/set-datetime-trained

set datetime trained when teaching
treehouselabs · Jan 6, 2015 · b770d39 · b770d39
2 parents a3e2779 + a722782
commit b770d39
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/FM/ClassificationBundle/Classifier/ResultPersistingClassifier.php b/src/FM/ClassificationBundle/Classifier/ResultPersistingClassifier.php
@@ -94,6 +94,10 @@ protected function persistResult($input, $output, $confidence, $expected = null)
         $classifyResult->setClassifier($this->classifierName);
         $classifyResult->setExpected($expected);
 
+        if (null !== $expected) {
+            $classifyResult->setDatetimeTrained(new \DateTime());
+        }
+
         // if we have an existing record, update it's score, otherwise insert
         if ($existingClassifyResult = $this->doctrine->getRepository('FMClassificationBundle:ClassifyResult')->findOneByHash($classifyResult->getHash())) {
             $existingClassifyResult->setOutput($output);
@@ -134,6 +138,8 @@ public function teach($input, $expected)
                 $classifyResult->incrementWeight();
 
                 $classifyResult->setScore(null);
+
+                $classifyResult->setDatetimeTrained(new \DateTime());
             }
 
             $this->doctrine->getManager()->flush($classifyResults);