fairer comparison

online-ml · Oct 5, 2023 · 4cb9795 · 4cb9795
1 parent 9a5a2e5
commit 4cb9795
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python_baseline.py b/python_baseline.py
@@ -3,14 +3,14 @@
 from time import time
 
 scores = []
-start = time()
 hst = anomaly.HalfSpaceTrees(
     n_trees=50,
     height=6,
     window_size=1000,
 )
-
-for x, _ in datasets.CreditCard():
+dataset = [x for x, _ in datasets.CreditCard()]
+start = time()
+for x in dataset:
     score = hst.score_one(x)
     scores.append(score)
     hst.learn_one(x)