Machine Learning, Data Science, Testing, Software Engineering.
like this: from yellowbrick.classifier import ROCAUC visualizer = ROCAUC(model, classes=classes) visualizer.fit(X_train, y_train) visualizer.score(X_test, y_test) visualizer.show() The class prediction error plot is done like this: from yellowbrick.classifier import ClassPredictionError visualizer