fix index error train model

XxRemsteelexX · Aug 2, 2024 · 92ef071 · 92ef071
1 parent f25065b
commit 92ef071
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train_model.py b/train_model.py
@@ -97,5 +97,5 @@
             model = model
         )
         with open("slice_output.txt", "a") as f:
-        print(f"{col}: {slicevalue}, Count: {count:,}", file=f)
-        print(f"Precision: {p:.4f} | Recall: {r:.4f} | F1: {fb:.4f}", file=f)
+            print(f"{col}: {slicevalue}, Count: {count:,}", file=f)
+            print(f"Precision: {p:.4f} | Recall: {r:.4f} | F1: {fb:.4f}", file=f)