sklearn 中 predict 方法和 predict_proba 方法的区别和使用
一、predict 和 predict_proba的概念和区别
1、predict和predict_proba都是用于模型的预测。
2、predict返回的是一个预测的值,predict_proba返回的是对于预测为各个类别的概率。
3、predict_proba返回的是一个 n 行 k 列的数组,n 表示测试集中样本的个数, 第 i 行 j列的数值是模型预测 第 i 个预测样本为某个标签的概率,并且每一行的概率和为1。
二、举例
1、predict
上图是对数据进行模型训练
从上图中可以看出predict方法可以对测试集test_x进行预测,而返回值则是0或1,表示预测成功或预测失败。
上图则是通过predict_proba方法对test_x进行预测,返回值为array数组,其中每行都代表一个样本,第一个值表示预测为0的概率,第二个值表示预测为1的概率,并且每行的概率值之和都为1。