from sklearn.metrics import confusion_matrix
y_true = [2, 0, 2, 2, 0, 1]
y_pred = [0, 0, 2, 2, 0, 2]
Когда я запускаю confusion_matrix(y_true, y_pred)
, результат будет следующим:
array([[2, 0, 0],
[0, 0, 1],
[1, 0, 2]], dtype=int64)
Имейте в виду, что результат для этого конкретного ввода - матрица 3x3, поэтому для этого случая вам понадобитсясписок с тремя именами для столбцов и индекса.Вы можете поместить результат непосредственно в Dataframe следующим образом:
pd.DataFrame(confusion_matrix(y_true, y_pred),columns=['column 1','column 2','column 3'], index=['index 1', 'index 2','index 3'])
, что дает следующий результат:
column 1 column 2 column 3
index 1 2 0 0
index 2 0 0 1
index 3 1 0 2