Skip to content

Commit 39363c3

Browse files
FIX - Fix implementation error in BinaryClassificationController, improve theoretical test notebook
1 parent 0be19bc commit 39363c3

File tree

2 files changed

+477
-327
lines changed

2 files changed

+477
-327
lines changed

mapie/risk_control_draft.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,9 +61,9 @@ def calibrate(self, X_calibrate: ArrayLike, y_calibrate: ArrayLike) -> None:
6161

6262
if self._risk.higher_is_better:
6363
risks_per_threshold = 1 - risks_per_threshold
64-
alpha = self._target_level
65-
else:
6664
alpha = 1 - self._target_level
65+
else:
66+
alpha = self._target_level
6767

6868
valid_thresholds_index = ltt_procedure(
6969
risks_per_threshold,

0 commit comments

Comments
 (0)