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。