Fix experiments-count cli parameter

soldag · Dec 6, 2016 · 179076a · 179076a
1 parent 07cadf1
commit 179076a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cli.py b/cli.py
@@ -25,8 +25,8 @@
 
     # Run specified number of experiments
     evaluation_data = []
-    experiments_count = args.experiments_count
-    for i in range(1, experiments_count):
+    experiments_count = int(args.experiments_count)
+    for i in range(0, experiments_count):
         tp, tn, fp, fn = run_pipeline(status_updates, args.classifier_type)
         evaluation_data.append([i, tp, tn, fp, fn, (tp + tn) / (tp + tn + fp + fn), tp / (tp + fp), tp / (tp + fn)])