[scikit-learn] Plot Cross-validated ROCs for multi-class classification problem

serafim loukas seralouk at hotmail.com
Fri Jul 20 05:35:18 EDT 2018


Dear Scikit-learn community,


I have a 3 class classification problem and I would like to plot the average ROC across Folds.
There is an example in scikit-learn website but only for binary classification problems (http://scikit-learn.org/stable/auto_examples/model_selection/plot_roc_crossval.html).

I want to do the same but in the case of 3 classes. I have tried to use `clf = OneVsRestClassifier(LinearDiscriminantAnalysis())`but I am having a hard time to make it work.


Any help would be appreciated,
Makis
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mail.python.org/pipermail/scikit-learn/attachments/20180720/f648fd04/attachment.html>


More information about the scikit-learn mailing list