mirror of
https://github.com/Doctorado-ML/STree.git
synced 2025-08-17 00:16:07 +00:00
Ricardo Montañana
044918f834
Create impurity function in Stree (consistent name, same criteria as other splitter parameter) Create test for the new function Update init test Update test splitter parameters Rename old impurity function to partition_impurity
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
import setuptools
|
|
|
|
__version__ = "0.9rc5"
|
|
__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>=0.23.0", "numpy", "ipympl"],
|
|
test_suite="stree.tests",
|
|
zip_safe=False,
|
|
)
|