From 82bc196dab434f4b1449478727316185145dab40 Mon Sep 17 00:00:00 2001 From: sheffer21 Date: Sun, 19 Apr 2020 11:47:03 +0300 Subject: [PATCH] fix --- MachineLearner/Main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/MachineLearner/Main.py b/MachineLearner/Main.py index bf48d0a..442a9f1 100644 --- a/MachineLearner/Main.py +++ b/MachineLearner/Main.py @@ -25,7 +25,7 @@ def main(outSourcedLogger=None): # Train the model - classificationAnalyzer = ClassificationResultAnalyzer(logger) + classificationAnalyzer = ClassificationResultAnalyzer(logger, 3) classification_model = ModelTrainer(logger, 3, lambda x: classifiers.classify_3classes(x, Threshold), "3_Classes_Training_with_threshold_5", MAX_LEN, epochs, batch_size,