ROCAUC Multi-class Visualizer

9 views
Skip to first unread message

Patrick Boada

unread,
Jun 15, 2021, 7:39:31 PM6/15/21
to Yellowbrick
Hi Yellowbrick,

I am running multi classification machine learning algorithms for analysis right now and I would love to use your ROCAUC function.

It works well in my jupyter notebook, however I cannot get it to save when I run the program in the command line.

Could you clarify if it is possible to save the ROCAUC plot to png?

Thank you,

Patrick

Lawrence Gray

unread,
Jun 17, 2021, 3:32:27 PM6/17/21
to Yellowbrick
Hi Patrick,  Thank you so very much for using Yellowbrick.

Have you tried adding the outpath parameter to the show method?
visualizer.show(outpath='rocauc.png')

Cheers,
Larry

Reply all
Reply to author
Forward
0 new messages