Machine Learning, Data Science, Testing, Software Engineering.
and support scores for the model for each class in the target variable. A confusion matrix is created like this: from yellowbrick.classifier import ConfusionMatrix visualizer = ConfusionMatrix(model, classes=classes) visualizer.score(X_test, y_test) visualizer.show() The ROC/AUC plot is created