gpt4 book ai didi

python - 我如何知道我的预测分类的标签?

转载 作者:行者123 更新时间:2023-11-30 09:43:22 25 4
gpt4 key购买 nike

我使用文本分类在 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_ 将按照分类器在分类器对象中存储的顺序为您提供分类器评分的标签。这应该与你已经得到的输出的顺序相同,尽管我会通过对你的预测进行一些抽查来确定这一点

关于python - 我如何知道我的预测分类的标签?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56080782/

25 4 0