Как я могу узнать метки моей предсказанной классификации? - PullRequest
0 голосов
/ 10 мая 2019

Я обучил свой классификатор на 3 диалектах, используя текстовую классификацию.И это были матрица и точность путаницы:

матрица путаницы

[[27  6  0 16]
 [ 5 18  0 21]
 [ 1  3  6  9]
 [ 0  0  0 48]]

Точность

[0.81818182 0.66666667 1.         0.5106383 ]

Как узнать, какая строка в матрице беспорядка и какой элемент вТочность принадлежит к тому, какой диалект у меня есть?Я предоставил обучающие данные классификатору со следующими метками:

Egyptian
Sudan
Iraqi
Jordan

Вот код, я использовал RandomForestClassifier:

from sklearn.ensemble import RandomForestClassifier
classifier = RandomForestClassifier(n_estimators=1000, random_state=0)  
classifier.fit(X, labels)  
test_pred = classifier.predict(y)
precision_score(labels_test,test_pred,average=None)

вывод:

array([0.91024735, 0.94929397, 0.98622273, 0,95343322])

1 Ответ

1 голос
/ 10 мая 2019

classifier.classes_ даст вам метки, по которым классификатор получает баллы в том порядке, в котором они хранятся в объекте классификатора.Это должно быть в том же порядке, что и результаты, которые вы уже получили, хотя я бы проверил это с некоторой выборочной проверкой ваших прогнозов, чтобы быть уверенным

...