Skip to content

Commit e661adb

Browse files
MTN - Clarify code
1 parent 1f18795 commit e661adb

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

mapie/risk_control_draft.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -56,17 +56,20 @@ def calibrate(self, X_calibrate: ArrayLike, y_calibrate: ArrayLike) -> None:
5656
) for predictions in predictions_per_threshold]
5757
)
5858

59+
risks_per_threshold = risks_and_eff_sizes[:, 0]
60+
eff_sample_sizes_per_threshold = risks_and_eff_sizes[:, 1]
61+
5962
if self._risk.higher_is_better:
60-
risks_and_eff_sizes[:, 0] = 1 - risks_and_eff_sizes[:, 0]
63+
risks_per_threshold = 1 - risks_per_threshold
6164
alpha = self._target_level
6265
else:
6366
alpha = 1 - self._target_level
6467

6568
valid_thresholds_index = ltt_procedure(
66-
risks_and_eff_sizes[:, 0],
69+
risks_per_threshold,
6770
np.array([alpha]),
6871
self._delta,
69-
risks_and_eff_sizes[:, 1],
72+
eff_sample_sizes_per_threshold,
7073
True,
7174
)
7275
self.valid_thresholds = self._thresholds[valid_thresholds_index[0]]

0 commit comments

Comments
 (0)