Right now, if you have probability set to True for a learner, you will not be able to use learning_curve() since it does not handle probabilities.
There are 2 solutions here:
- Show a warning and run argmax over the probabilities to get the winning index and run the rest of the learning curve code as normal.
- Raise an error saying that this is not supported.
Thoughts?