[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