机器学习预测的概率如何表示?
机器学习预测的概率如何表示?
调用predict_prob
函数可以显示出预测每一类的概率矩阵
classifier = SGDClassifier(loss='log', n_iter_no_change=100)
#注意这里loss的值是log,是概率模型,如果设置为默认的hinge,就没有predict_proba这个函数
#所以大家看看说明,看看是不是概率模型,多试试
classifier.fit(train_features , train_labels)
pred_proba = classifier.predict_proba(test_features)
print(pred_proba)
结果是这种形式:
[[0.56651809 0.43348191]
[0.15598162 0.84401838]
[0.86852502 0.13147498]]
这里注意一下,第一列指的是预测标签最小的那一类的,换句话说就算你的特征对应的标签是这样的:
y = np.array([3, 2])
第一列预测的还是标签为2的这一类的概率