Correct column name.

master
junos 2023-05-18 18:47:56 +02:00
parent fa45b30955
commit 45441c288d
1 changed files with 2 additions and 2 deletions

View File

@ -182,7 +182,7 @@ def run_all_regression_models(
) -> pd.DataFrame: ) -> pd.DataFrame:
metrics = ["r2", "neg_mean_absolute_error", "neg_root_mean_squared_error"] metrics = ["r2", "neg_mean_absolute_error", "neg_root_mean_squared_error"]
test_metrics = ["test_" + metric for metric in metrics] test_metrics = ["test_" + metric for metric in metrics]
scores = pd.DataFrame(columns=["method", "metric", "max", "nanmedian"]) scores = pd.DataFrame(columns=["method", "test_metric", "max", "nanmedian"])
# Validate models # Validate models
dummy_regr = DummyRegressor(strategy="mean") dummy_regr = DummyRegressor(strategy="mean")
@ -427,7 +427,7 @@ def run_all_classification_models(
metrics = ["accuracy", "average_precision", "recall", "f1"] metrics = ["accuracy", "average_precision", "recall", "f1"]
test_metrics = ["test_" + metric for metric in metrics] test_metrics = ["test_" + metric for metric in metrics]
scores = pd.DataFrame(columns=["method", "metric", "max", "mean"]) scores = pd.DataFrame(columns=["method", "test_metric", "max", "mean"])
dummy_class = DummyClassifier(strategy="most_frequent") dummy_class = DummyClassifier(strategy="most_frequent")