-
Notifications
You must be signed in to change notification settings - Fork 135
Description
Hi team,
I noticed that VennAbersCalibrator relies internally on the VennAbers class. The underlying VennAbers.predict_proba method supports the p0_p1_output parameter, which allows returning the (p0, p1) pair instead of the averaged calibrated probability.
In the current MAPIE implementation, however, this parameter is not exposed in VennAbersCalibrator.predict_proba. It appears that p0_p1_output is effectively fixed to False internally, and users cannot control this behavior through the public API.
Why this matters
In some workflows, it can be useful to access the raw (p0, p1) outputs rather than only the averaged calibrated probability. For example, the interval (p0, p1) may be reported alongside the calibrated probability as a measure of uncertainty.
Since the underlying VennAbers implementation already supports this functionality, exposing the p0_p1_output parameter in the MAPIE wrapper would increase flexibility without altering default behavior.
Would it be possible to expose this parameter in VennAbersCalibrator.predict_proba?
Please let me know if I have misunderstood the intended design or missed something in the implementation.
Thanks for your work on MAPIE!