The PRCurve Visualizer has an issue in the .fit() method for multi-class classifier models

DistrictDataLabs / yellowbrick

Visual analysis and diagnostic tools to facilitate machine learning model selection.

http://www.scikit-yb.org/

Apache License 2.0

4.3k stars 559 forks source link

The PRCurve Visualizer has an issue in the .fit() method for multi-class classifier models #1191

Open pdamodaran opened 3 years ago

pdamodaran commented 3 years ago

Describe the bug In the PRCurve Visualizer, the multi-class classifier models get wrapped up in a OneVsRestClassifier but this classifier gets ignored in the call to super and does not get fitted as a result. This results in an error in rendering the visualizer as the num_class parameter is not set

To Reproduce

This was discovered through Issue #1190.