Remove min_data hyperparameter

This commit is contained in:
2022-11-18 23:14:41 +01:00
parent 0a7c56f339
commit 03fd1fec9e
2 changed files with 2 additions and 7 deletions

View File

@@ -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"]

View File

@@ -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_ == [