enables serial regression training

jim-schwoebel · Aug 14, 2020 · 2d79d49 · 2d79d49
1 parent fd32c76
commit 2d79d49
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/training/model.py b/training/model.py
@@ -505,6 +505,8 @@ def plot_regressor(regressor, classes, X_test, y_test):
 		for i in range(int(classnum)):
 			classes.append(sys.argv[i+5])
 	else:
+
+		classnum=1
 		mtype=sys.argv[1]
 		csvfile=sys.argv[2]
 		classes=[sys.argv[3]]