mirror of
https://github.com/Doctorado-ML/STree.git
synced 2025-08-15 23:46:02 +00:00
147dad684c
* Add Hyperparameters description to README Comment get_subspace method Add environment info for binder (runtime.txt) * Complete source comments Change docstring type to numpy update hyperameters table and explanation * Fix problem with zero weighted samples Solve WARNING: class label x specified in weight is not found with a different approach * Allow update of scikitlearn to latest version
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
import setuptools
|
|
|
|
__version__ = "1.0rc1"
|
|
__author__ = "Ricardo Montañana Gómez"
|
|
|
|
|
|
def readme():
|
|
with open("README.md") as f:
|
|
return f.read()
|
|
|
|
|
|
setuptools.setup(
|
|
name="STree",
|
|
version=__version__,
|
|
license="MIT License",
|
|
description="Oblique decision tree with svm nodes",
|
|
long_description=readme(),
|
|
long_description_content_type="text/markdown",
|
|
packages=setuptools.find_packages(),
|
|
url="https://github.com/doctorado-ml/stree",
|
|
author=__author__,
|
|
author_email="ricardo.montanana@alu.uclm.es",
|
|
keywords="scikit-learn oblique-classifier oblique-decision-tree decision-\
|
|
tree svm svc",
|
|
classifiers=[
|
|
"Development Status :: 4 - Beta",
|
|
"License :: OSI Approved :: MIT License",
|
|
"Programming Language :: Python :: 3.8",
|
|
"Natural Language :: English",
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
|
"Intended Audience :: Science/Research",
|
|
],
|
|
install_requires=["scikit-learn", "numpy", "ipympl"],
|
|
test_suite="stree.tests",
|
|
zip_safe=False,
|
|
)
|