From 9e620a07039153372d2d396d28970fb94623c43a Mon Sep 17 00:00:00 2001
From: Yisen Jin <66319281+yjin07@users.noreply.github.com>
Date: Sun, 29 Oct 2023 00:44:19 +0000
Subject: [PATCH] correct sklearn_NB.var_
---
numpy_ml/tests/test_naive_bayes.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/numpy_ml/tests/test_naive_bayes.py b/numpy_ml/tests/test_naive_bayes.py
index b8cf2d9..24da96f 100644
--- a/numpy_ml/tests/test_naive_bayes.py
+++ b/numpy_ml/tests/test_naive_bayes.py
@@ -42,13 +42,13 @@ def test_GaussianNB(N=10):
np.testing.assert_almost_equal(jointi, jointi_mine)
- n_jk = -0.5 * np.sum(np.log(2.0 * np.pi * sklearn_NB.sigma_[j, :] + eps))
+ n_jk = -0.5 * np.sum(np.log(2.0 * np.pi * sklearn_NB.var_[j, :] + eps))
n_jk_mine = -0.5 * np.sum(np.log(2.0 * np.pi * P["sigma"][j] + eps))
np.testing.assert_almost_equal(n_jk_mine, n_jk)
n_jk2 = n_jk - 0.5 * np.sum(
- ((X_test - sklearn_NB.theta_[j, :]) ** 2) / (sklearn_NB.sigma_[j, :]), 1
+ ((X_test - sklearn_NB.theta_[j, :]) ** 2) / (sklearn_NB.var_[j, :]), 1
)
n_jk2_mine = n_jk_mine - 0.5 * np.sum(
@@ -63,7 +63,7 @@ def test_GaussianNB(N=10):
np.testing.assert_almost_equal(P["prior"], sklearn_NB.class_prior_)
np.testing.assert_almost_equal(P["mean"], sklearn_NB.theta_)
- np.testing.assert_almost_equal(P["sigma"], sklearn_NB.sigma_)
+ np.testing.assert_almost_equal(P["sigma"], sklearn_NB.var_)
np.testing.assert_almost_equal(
sklearn_NB._joint_log_likelihood(X_test),
NB._log_posterior(X_test),