Merge pull request #10 from sverchkov/show-err-name

Show metric name in measure_model_err output
carpentries-incubator · Jun 20, 2024 · 2044a11 · 2044a11
2 parents 79ca967 + 7459f1e
commit 2044a11
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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