logging improved

grro · Nov 4, 2022 · 71a3cfe · 71a3cfe
1 parent bc5b00f
commit 71a3cfe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pvpower/forecast.py b/pvpower/forecast.py
@@ -35,7 +35,7 @@ def select_best_estimator(self, train_log: TrainSampleLog, num_rounds: int = 5)
         report += "VARIANT ................................. SCORE\n"
         for variant, vectorizer in vectorizer_map.items():
             estimator = Estimator(vectorizer)
-            median_report = estimator.test(samples, rounds=10)
+            median_report = estimator.test(samples, rounds=num_rounds)
             score_str = str(round(median_report.score, 1))
             report += variant + " " + "".join(["."] * (45 - (len(variant)+len(score_str)))) + " " + score_str + "\n"
             if median_report.score < lowest_score: