confusion_matrix()
Матрица ошибок классификации.
Сигнатура
confusion_matrix(y_true, y_pred)Возвращает таблицу, где строки — истинные классы, столбцы — предсказанные. По диагонали стоят верные ответы, вне диагонали — ошибки. Помогает понять, какие классы путаются.
from sklearn.metrics import confusion_matrix
print(confusion_matrix(y_test, y_pred))