RedFlag metric

uzh-dqbm-cmi · Aug 14, 2023 · fdeb72e · fdeb72e
1 parent 8d25434
commit fdeb72e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/mridle/experiment/tuner.py b/src/mridle/experiment/tuner.py
@@ -104,7 +104,7 @@ def hyperopt_objective(cls, params, model, x_train, y_train, scoring_fn: str, id
 
             model_copy = model_copy.fit(x_train_cv, y_train_cv)
 
-            if scoring_fn not in ['mse', 'mae', 'rmse', 'mape']:
+            if scoring_fn not in ['mse', 'mae', 'rmse', 'mape', 'redflag']:
                 y_proba_preds = model_copy.predict_proba(x_test_cv)
                 y_proba_preds = np.clip(y_proba_preds, 1e-5, 1 - 1e-5)
                 if y_proba_preds.shape[1] == 2: