From eb71f96623fb63e8e4cbf387d7f0d25f059bb24e Mon Sep 17 00:00:00 2001 From: Mariya Lysenkova Wiklander Date: Thu, 7 Nov 2024 14:09:38 +0100 Subject: [PATCH] Fixed regression bug --- src/conformist/validation_trial.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/conformist/validation_trial.py b/src/conformist/validation_trial.py index 0a97fd4..11f6538 100644 --- a/src/conformist/validation_trial.py +++ b/src/conformist/validation_trial.py @@ -91,7 +91,8 @@ def mean_set_sizes_by_class(self, class_names): def mean_fnrs_by_class(self, class_names): fnr_dicts = [] for run in self.runs: - fnr_dicts.append(run.mean_fnrs_by_class(class_names)) + fnr_dicts.append(run.mean_fnrs_by_class(run.prediction_sets, + class_names)) means = {} for class_name in class_names: