Merge branch 'ahnentafel' of https://github.com/online-ml/light-river …

…into ahnentafel
online-ml · Oct 5, 2023 · 126061a · 126061a
2 parents aa9e2ac + 4cb9795
commit 126061a
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)