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

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