Update doc and version 1.30 (#55)

* Add complete classes counts to node and tests

* Implement optimized predict and new predict_proba

* Add predict_proba test

* Add python 3.10 to CI

* Update version number and documentation
This commit is contained in:
Ricardo Montañana Gómez
2022-10-21 13:31:59 +02:00
committed by GitHub
parent 2f6ae648a1
commit c37f044e3a
6 changed files with 53 additions and 15 deletions

View File

@@ -7,9 +7,8 @@ def readme():
return f.read()
def get_data(field):
def get_data(field, file_name="__init__.py"):
item = ""
file_name = "_version.py" if field == "version" else "__init__.py"
with open(os.path.join("stree", file_name)) as f:
for line in f.readlines():
if line.startswith(f"__{field}__"):
@@ -21,9 +20,14 @@ def get_data(field):
return item
def get_requirements():
with open("requirements.txt") as f:
return f.read().splitlines()
setuptools.setup(
name="STree",
version=get_data("version"),
version=get_data("version", "_version.py"),
license=get_data("license"),
description="Oblique decision tree with svm nodes",
long_description=readme(),
@@ -46,7 +50,7 @@ setuptools.setup(
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Intended Audience :: Science/Research",
],
install_requires=["scikit-learn", "mufs"],
install_requires=get_requirements(),
test_suite="stree.tests",
zip_safe=False,
)