From 7459f1efb25ff96d08be693f2b10d4d94a7d7eb3 Mon Sep 17 00:00:00 2001 From: Yuriy Sverchkov Date: Tue, 15 Aug 2023 13:58:33 -0500 Subject: [PATCH] Show metric name in measure_model_err output --- code/regression_predict_sklearn.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/code/regression_predict_sklearn.py b/code/regression_predict_sklearn.py index b631dd4..c318d72 100644 --- a/code/regression_predict_sklearn.py +++ b/code/regression_predict_sklearn.py @@ -108,9 +108,9 @@ def measure_model_err(y: Union[np.ndarray, pd.Series], baseline_pred: Union[floa # Create a DataFrame to store the error values errors_df = pd.DataFrame({ - 'Baseline Error': [baseline_err], - 'Train Error': [train_err], - 'Test Error': [test_err] + f'Baseline {metric}': [baseline_err], + f'Train {metric}': [train_err], + f'Test {metric}': [test_err] }) return errors_df