sklearn_logistic.predict_proba函数介绍

函数介绍

sklearn_logistic.predict_proba 是 scikit-learn 中 LogisticRegression 模型对象的一个方法，用于获取模型对输入样本属于各个类别的概率预测值。

具体来说，predict_proba 方法接受输入样本集合作为参数，然后返回一个数组，数组的每一行对应输入样本的预测结果，每一列对应一个类别，数组中的每个元素表示对应样本属于该类别的概率。

在二分类问题中，predict_proba 返回的数组的第一列表示样本属于第一个类别的概率，第二列表示样本属于第二个类别的概率。因此，如果你想获取模型对测试集中样本为正例的预测概率，你可以使用 [:,1] 来提取出第二列的概率值。

方法

例如，如果你有一个 LogisticRegression 模型对象 model，你可以使用以下方式调用 predict_proba 方法：

y_proba = model.predict_proba(X_test)
y_proba1 = model.predict_proba(X_test)[:,1]  #正例的预测概率

这将会给你一个数组 y_proba，其中每一行表示一个测试样本，每一列表示一个类别的概率值。