我怎么知道我的预测分类的标签?
How can I know the labels of my predicted classification?
我使用文本分类训练了 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])
classifier.classes_
将按照标签在分类器对象中的存储顺序为您提供分类器评分的标签。这应该与你已经得到的输出的顺序相同,尽管我会用你的一些 spot-checking 的预测来验证这一点
我使用文本分类训练了 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])
classifier.classes_
将按照标签在分类器对象中的存储顺序为您提供分类器评分的标签。这应该与你已经得到的输出的顺序相同,尽管我会用你的一些 spot-checking 的预测来验证这一点