Sklearn弹性网络逻辑回归（SGDClassifier）没有返回概率

Question

在sklearn中，当使用SGDCLassifier进行弹性网络逻辑回归时，predict_proba函数返回与predict函数相同的结果。

AKA下面的代码（具有X和y预测器和分别二进制标记）返回True：

EN = sklearn.linear_model.SGDClassifier(loss='log', penalty='elasticnet', 
             alpha=0.0001, l1_ratio=0.15) 
EN.fit(X[train], y[train]) 
numpy.all(EN.predict(X[test]) == EN.predict_proba(X[test])[:,1]) 

如何获取概率值？

Answer 1

似乎sklearn版本是问题。您需要升级到0.18.2。

实施例使用虹膜数据：

from sklearn.datasets import load_iris 
from sklearn.linear import model.SGDClassifier 
from sklearn.model_selection import train_test_split 
import numpy 
import sklearn 

data = load_iris() 
x = data.data 
y = data.target 
X_train, X_test, y_train, y_test = train_test_split(x, y, test_size = 0.3, random_state = 42) 

EN = SGDClassifier(loss='log', penalty='elasticnet', alpha=0.0001, l1_ratio=0.15) 

EN.fit(X_train, y_train) 

numpy.all(EN.predict(X_test) == EN.predict_proba(X_test)[:,1]) 
sklearn.__version__ 

结果

False 

'0.18.2' 

因此，与sklearn 0.18.2它工作正常。