Skip to content

Commit f524038

Browse files
committed
rename list_params input
1 parent 5c398df commit f524038

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

mapie/risk_control/binary_classification.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ class BinaryClassificationController:
9090
"fpr" for false positive rate.
9191
- A custom instance of BinaryClassificationRisk object
9292
93-
predict_params : NDArray, default=np.linspace(0, 0.99, 100)
93+
list_params : NDArray, default=np.linspace(0, 0.99, 100)
9494
The set of parameters (noted λ in [1]) to consider for controlling the risk (or performance).
9595
When `predict_function` is a `predict_proba` method, the shape is (n_params,)
9696
and the parameter values are used to threshold the probabilities.
@@ -172,7 +172,7 @@ def __init__(
172172
best_predict_param_choice: Union[
173173
Literal["auto"], Risk_str, BinaryClassificationRisk
174174
] = "auto",
175-
predict_params: NDArray = np.linspace(0, 0.99, 100),
175+
list_params: NDArray = np.linspace(0, 0.99, 100),
176176
):
177177
self.is_multi_risk = self._check_if_multi_risk_control(risk, target_level)
178178
self._predict_function = predict_function
@@ -199,9 +199,9 @@ def __init__(
199199
best_predict_param_choice
200200
)
201201

202-
self._predict_params = predict_params
202+
self._predict_params = list_params
203203
self.is_multi_dimensional_param = self._check_if_multi_dimensional_param(
204-
predict_params
204+
self._predict_params
205205
)
206206

207207
self.valid_predict_params: NDArray = np.array([])

0 commit comments

Comments
 (0)