Skip to content

Learning curve won't work for probabilistic classifiers #648

@desilinguist

Description

@desilinguist

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:

  1. 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.
  2. Raise an error saying that this is not supported.

Thoughts?

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions