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),