From 7d075219d42cf5c8cca62c6c9804dba8c04907cc Mon Sep 17 00:00:00 2001 From: Balazs Kegl Date: Tue, 2 Nov 2021 12:53:22 +0100 Subject: [PATCH] Fix blending decoupled cv (#282) * Update __init__.py Exposing ClassifierBaseScoreType so kits can use it. * Update testing.py --- rampwf/utils/testing.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/rampwf/utils/testing.py b/rampwf/utils/testing.py index d7f5775d..37aba8a3 100644 --- a/rampwf/utils/testing.py +++ b/rampwf/utils/testing.py @@ -222,7 +222,8 @@ def blend_submissions(submissions, ramp_kit_dir='.', ramp_data_dir='.', foldwise_best_predictions_test_list = [] for fold_i, valid_is in enumerate(valid_is_list): print_title('CV fold {}'.format(fold_i)) - ground_truths_valid = problem.Predictions(y_true=y_train[valid_is]) + ground_truths_valid = problem.Predictions( + y_true=y_train, fold_is=valid_is) predictions_valid_list = [] predictions_test_list = [] for submission in submissions: @@ -241,7 +242,7 @@ def blend_submissions(submissions, ramp_kit_dir='.', ramp_data_dir='.', problem, data_path=ramp_data_dir, input_path=fold_output_path, suffix='test') predictions_valid = problem.Predictions( - y_pred=y_pred_train[valid_is]) + y_pred=y_pred_train, fold_is=valid_is) predictions_valid_list.append(predictions_valid) predictions_test = problem.Predictions(y_pred=y_pred_test) predictions_test_list.append(predictions_test)