Permalink
Browse files

COSMIT in naive_bayes

  • Loading branch information...
larsmans committed Oct 28, 2011
1 parent f997740 commit 2e42f8950171d99f067a246fa0c175c23b325c5b
Showing with 1 addition and 2 deletions.
  1. +1 −2 sklearn/naive_bayes.py
View
@@ -61,8 +61,7 @@ def predict(self, X):
Predicted target values for X
"""
jll = self._joint_log_likelihood(X)
- y_pred = self._classes[np.argmax(jll, axis=1)]
- return y_pred
+ return self._classes[np.argmax(jll, axis=1)]
def predict_log_proba(self, X):
"""

0 comments on commit 2e42f89

Please sign in to comment.