diff --git a/bayesclass/bayesclass.py b/bayesclass/bayesclass.py index 6264919..49acd6e 100644 --- a/bayesclass/bayesclass.py +++ b/bayesclass/bayesclass.py @@ -324,11 +324,10 @@ class KDB(BayesBase): class AODE(BayesBase, BaseEnsemble): - def __init__(self, min_data=30, show_progress=False, random_state=None): + def __init__(self, show_progress=False, random_state=None): super().__init__( show_progress=show_progress, random_state=random_state ) - self.min_data = min_data def _check_params(self, X, y, kwargs): expected_args = ["class_name", "features"] diff --git a/bayesclass/tests/test_AODE.py b/bayesclass/tests/test_AODE.py index 9d2e602..4bd1d00 100644 --- a/bayesclass/tests/test_AODE.py +++ b/bayesclass/tests/test_AODE.py @@ -2,8 +2,6 @@ import pytest import numpy as np from sklearn.datasets import load_iris from sklearn.preprocessing import KBinsDiscretizer -from matplotlib.testing.decorators import image_comparison -from matplotlib.testing.conftest import mpl_test_settings from bayesclass import AODE @@ -26,11 +24,9 @@ def test_AODE_default_hyperparameters(data, clf): # Test default values of hyperparameters assert not clf.show_progress assert clf.random_state is None - assert clf.min_data == 30 - clf = AODE(show_progress=True, random_state=17, min_data=3) + clf = AODE(show_progress=True, random_state=17) assert clf.show_progress assert clf.random_state == 17 - assert clf.min_data == 3 clf.fit(*data) assert clf.class_name_ == "class" assert clf.features_ == [