feat(Evaluation): created a unified evaluation framework for both regression / classification

This commit is contained in:
Mark Aron Szulyovszky
2021-12-14 21:25:43 +01:00
parent 1ef314c034
commit beb281fc3a
4 changed files with 46 additions and 50 deletions
+1 -2
View File
@@ -10,9 +10,8 @@ def walk_forward_train_test(
y: pd.Series,
window_size: int,
retrain_every: int
) -> tuple[list[SKLearnModel], list[float]]:
) -> tuple[pd.Series, pd.Series]:
print("Training: ", model_name)
predictions = pd.Series(index=y.index)
models = pd.Series(index=y.index)