predict_proba如何查看类别概率对应的类别名称
时间: 2024-02-02 14:04:16 浏览: 93
sklearn的predict_proba使用说明
在使用predict_proba函数预测样本的类别概率时,可以通过调用模型的classes_属性来获取类别名称。该属性返回一个数组,包含了模型训练时所使用的所有类别名称。可以通过以下方式来查看类别概率对应的类别名称:
```python
# 假设已经定义好了模型变量为model,输入数据为X
import numpy as np
# 预测样本的概率
probas = model.predict_proba(X)
# 获取类别名称
class_names = np.array(model.classes_)
# 打印每个样本的预测结果
for i in range(len(X)):
print("样本 %d 的预测结果:" % i)
for j in range(len(class_names)):
print("%s 的概率为:%.2f" % (class_names[j], probas[i][j]))
```
在上述代码中,我们首先通过model.predict_proba(X)获取样本的类别概率,然后通过model.classes_获取类别名称,最后使用循环打印每个样本的预测结果,包括每个类别的概率和类别名称。
阅读全文