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它工作正常。