FIX: Behavior of classes_ attr w/new BaseLibLinear

HapeMask · HapeMask · commit 3818dfca9095 · 2014-07-29T15:10:51.000-04:00
BaseLibLinear and derived classes no longer have a "classes_" attribute
until fit() has been called.
diff --git a/sklearn/linear_model/logistic.py b/sklearn/linear_model/logistic.py
@@ -520,6 +520,7 @@ def _log_reg_scoring_path(X, y, train, test, pos_class=None, Cs=10,
     log_reg = LogisticRegression(fit_intercept=fit_intercept)
     log_reg._enc = LabelEncoder()
     log_reg._enc.fit_transform([-1, 1])
+    log_reg.classes_ = log_reg._enc.classes_
 
     X_train = X[train]
     X_test = X[test]
@@ -920,6 +921,7 @@ def fit(self, X, y):
 
         self._enc = LabelEncoder()
         self._enc.fit(y)
+        self.classes_ = self._enc.classes_
 
         labels = self.classes_
         n_classes = len(labels)
