Update to use setuptools

bayesclass/__init__.py

@@ -1,5 +1,5 @@
-from .bayesclass import TAN, KDB, AODE
-from ._version import __version__
+# from .bayesclass import TAN, KDB, AODE
+from bayesclass._version import __version__
 
 __author__ = "Ricardo Montañana Gómez"
 __copyright__ = "Copyright 2020-2023, Ricardo Montañana Gómez"

bayesclass/bayesclass.py

@@ -86,7 +86,7 @@ class BayesBase(BaseEstimator, ClassifierMixin):
         --------
         >>> import numpy as np
         >>> import pandas as pd
-        >>> from bayesclass import TAN
+        >>> from bayesclass.bayesclass import TAN
         >>> features = ['A', 'B', 'C', 'D', 'E']
         >>> np.random.seed(17)
         >>> values = pd.DataFrame(np.random.randint(low=0, high=2,
@@ -142,7 +142,7 @@ class BayesBase(BaseEstimator, ClassifierMixin):
         --------
         >>> import numpy as np
         >>> import pandas as pd
-        >>> from bayesclass import TAN
+        >>> from bayesclass.bayesclass import TAN
         >>> features = ['A', 'B', 'C', 'D', 'E']
         >>> np.random.seed(17)
         >>> values = pd.DataFrame(np.random.randint(low=0, high=2,
@@ -373,4 +373,4 @@ class AODE(BayesBase, BaseEnsemble):
         dataset = pd.DataFrame(X, columns=self.features_, dtype="int16")
         for index, model in enumerate(self.models_):
             result[:, index] = model.predict(dataset).values.ravel()
-        return mode(result, axis=1).mode.ravel()
+        return mode(result, axis=1, keepdims=False).mode.ravel()

bayesclass/tests/test_AODE.py

@@ -4,7 +4,7 @@ from sklearn.datasets import load_iris
 from sklearn.preprocessing import KBinsDiscretizer
 
 
-from bayesclass import AODE
+from bayesclass.bayesclass import AODE
 from .._version import __version__
 
 

bayesclass/tests/test_KDB.py

@@ -6,7 +6,7 @@ from matplotlib.testing.decorators import image_comparison
 from matplotlib.testing.conftest import mpl_test_settings
 
 
-from bayesclass import KDB
+from bayesclass.bayesclass import KDB
 from .._version import __version__
 
 

bayesclass/tests/test_TAN.py

@@ -6,7 +6,7 @@ from matplotlib.testing.decorators import image_comparison
 from matplotlib.testing.conftest import mpl_test_settings
 
 
-from bayesclass import TAN
+from bayesclass.bayesclass import TAN
 from .._version import __version__
 
 

bayesclass/tests/test_common.py

@@ -2,7 +2,7 @@ import pytest
 
 from sklearn.utils.estimator_checks import check_estimator
 
-from bayesclass import TAN, KDB, AODE
+from bayesclass.bayesclass import TAN, KDB, AODE
 
 
 @pytest.mark.parametrize("estimator", [TAN(), KDB(k=2), AODE()])

bayesclass/tests/test_doctest.py

@@ -0,0 +1,7 @@
+import doctest
+import bayesclass
+
+
+def load_tests(loader, tests, ignore):
+    tests.addTests(doctest.DocTestSuite(bayesclass))
+    return tests