added accuracy test

flame-stream · Feb 27, 2019 · 823d053 · 823d053
1 parent cd6e40d
commit 823d053
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/examples/src/test/java/com/spbsu/flamestream/example/bl/text_classifier/LentaTest.java b/examples/src/test/java/com/spbsu/flamestream/example/bl/text_classifier/LentaTest.java
@@ -58,6 +58,7 @@
 
 import static com.spbsu.flamestream.example.bl.classifier.PredictorStreamTest.parseDoubles;
 import static java.util.stream.Collectors.toList;
+import static org.testng.Assert.assertTrue;
 
 public class LentaTest extends FlameAkkaSuite {
   private static final Logger LOGGER = LoggerFactory.getLogger(LentaTest.class);
@@ -179,7 +180,9 @@ public void partialFitTest() {
       //LOGGER.info("\n");
     }
 
-    LOGGER.info("Accuracy: {}", truePositives / testsize);
+    double accuracy = truePositives / testsize;
+    LOGGER.info("Accuracy: {}", accuracy);
+    assertTrue(accuracy >= 0.62);
   }
 
   @Test