BUG don't use deprecated attributes in GaussianNB.predict

larsmans · larsmans · commit f4eced300177 · 2012-01-09T15:53:27.000+01:00
diff --git a/sklearn/naive_bayes.py b/sklearn/naive_bayes.py
@@ -171,9 +171,9 @@ def _joint_log_likelihood(self, X):
         joint_log_likelihood = []
         for i in xrange(np.size(self._classes)):
             jointi = np.log(self.class_prior_[i])
-            n_ij = - 0.5 * np.sum(np.log(np.pi * self.sigma[i, :]))
-            n_ij -= 0.5 * np.sum(((X - self.theta[i, :]) ** 2) / \
-                                    (self.sigma[i, :]), 1)
+            n_ij = - 0.5 * np.sum(np.log(np.pi * self.sigma_[i, :]))
+            n_ij -= 0.5 * np.sum(((X - self.theta_[i, :]) ** 2) / \
+                                    (self.sigma_[i, :]), 1)
             joint_log_likelihood.append(jointi + n_ij)
         joint_log_likelihood = np.array(joint_log_likelihood).T
         return joint_log_likelihood
