small bug on train svm

SupervisedStylometry · Apr 30, 2024 · ae49458 · ae49458
1 parent d359bc8
commit ae49458
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/train_svm.py b/train_svm.py
@@ -69,8 +69,11 @@
     else:
         args.o = ''
 
-    svm["confusion_matrix"].to_csv(args.o+"confusion_matrix.csv")
-    svm["misattributions"].to_csv(args.o+"misattributions.csv")
+
+    if args.cross_validate is not None or (args.test_path is not None and not args.final):
+        svm["confusion_matrix"].to_csv(args.o+"confusion_matrix.csv")
+        svm["misattributions"].to_csv(args.o+"misattributions.csv")
+
     joblib.dump(svm["pipeline"], args.o+'mySVM.joblib')
 
     if args.final: