Участка путаницы матрица sklearn с несколькими ярлыками

Question

Я черчение матрицы путаницы в течение нескольких маркированы данных, где метки похожи:

label1: 1, 0, 0, 0

LABEL2: 0, 1, 0, 0

Метка3: 0, 0, 1, 0

label4: 0, 0, 0, 1

Я могу успешно классифицировать, используя приведенный ниже код. Мне нужна только помощь для построения матрицы смешения.

for i in range(4): 
     y_train= y[:,i] 
     print('Train subject %d, class %s' % (subject, cols[i])) 
     lr.fit(X_train[::sample,:],y_train[::sample]) 
     pred[:,i] = lr.predict_proba(X_test)[:,1] 

Я использовал следующий код для печати спутанности матрицы, но всегда возвращает 2X2 матрицу

prediction = lr.predict(X_train) 

print(confusion_matrix(y_train, prediction)) 

Answer 1

Я вижу это по-прежнему остается открытым вопрос в sklearn «s хранилище:

https://github.com/scikit-learn/scikit-learn/issues/3452

Однако были предприняты некоторые попытки его реализации. Из того же # 3452 выпуска нити:

https://github.com/Magellanea/scikit-learn/commit/514287c1d5dad2f0ab4918dc4da5cf7053fe6734#diff-b04acd877dd793f28ae7be13a999ed88R187

Вы можете проверить код, предложенный в функции и посмотреть, если это соответствует вашим потребностям.