From 5b29e82a05c079647e8dfb5f0fc561e034c42d70 Mon Sep 17 00:00:00 2001
From: Gael Varoquaux <gael.varoquaux@normalesup.org>
Date: Sun, 6 May 2012 23:26:30 +0200
Subject: [PATCH] Revert "ENH: avoid an underflow"
That was a bad idea: this underflow is happening because there is only
onlyone class
This reverts commit 65abbb674195d279608346af3e0c2b367f6e4b33.
---
sklearn/naive_bayes.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/sklearn/naive_bayes.py b/sklearn/naive_bayes.py
index 4e7345bf0f..eb29765efe 100644
--- a/sklearn/naive_bayes.py
+++ b/sklearn/naive_bayes.py
@@ -266,9 +266,7 @@ class BaseDiscreteNB(BaseNB):
elif self.fit_prior:
# empirical prior, with sample_weight taken into account
y_freq = Y.sum(axis=0)
- eps = np.finfo(1.).eps
- self.class_log_prior_ = (np.log(y_freq + eps)
- - np.log(y_freq.sum() + eps))
+ self.class_log_prior_ = np.log(y_freq) - np.log(y_freq.sum())
else:
self.class_log_prior_ = np.zeros(n_classes) - np.log(n_classes)
--
GitLab