Doctorado-ML/bayesclass
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/bayesclass.git synced 2025-08-19 09:35:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Begin with kDB classifier

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2022-11-14 14:03:45 +01:00
parent 6b2e60eba0
commit 1545ca62cf
4 changed files with 168 additions and 133 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
bayesclass/tests/test_bayesclass.py
View File

@@ -69,8 +69,7 @@ def test_TAN_classifier(data):
X = data[0]
y = data[1]
y_pred = clf.predict(X)
y = y.reshape(-1, 1)
assert y_pred.shape == (X.shape[0], 1)
assert y_pred.shape == (X.shape[0],)
assert sum(y == y_pred) == 147
@@ -103,8 +102,7 @@ def test_TAN_classifier_simple_init(data):
X = data[0]
y = data[1]
y_pred = clf.predict(X)
y = y.reshape(-1, 1)
assert y_pred.shape == (X.shape[0], 1)
assert y_pred.shape == (X.shape[0],)
assert sum(y == y_pred) == 147

5
bayesclass/tests/test_common.py
View File

@@ -7,4 +7,7 @@ from bayesclass import TAN
@pytest.mark.parametrize("estimator", [TAN()])
def test_all_estimators(estimator):
return check_estimator(estimator)
i = 0
for estimator, test in check_estimator(estimator, generate_only=True):
print(i := i + 1, test, "classes_")
# test(estimator)