Fix AttributeError in prediction.

lazywei · lazywei · commit a2774cb67676 · 2014-10-02T14:08:03.000+08:00
diff --git a/sklearn/multi_label/classifier_chain.py b/sklearn/multi_label/classifier_chain.py
@@ -25,7 +25,7 @@ class ClassifierChain(BaseEstimator):
 
     Attributes
     ----------
-    n_labels : int
+    n_labels_ : int
         How many labels are there in this model.
 
     classifiers_ : array
@@ -99,7 +99,7 @@ def predict(self, X):
         for clf in self.classifiers_:
             X = self._predict_and_chain(clf, X)
 
-        return X[:, -self.n_labels:]
+        return X[:, -self.n_labels_:]
 
     def _predict_and_chain(self, clf, X):
         return np.hstack((X, clf.predict(X).reshape(-1, 1)))
