mirror of
https://github.com/Doctorado-ML/STree.git
synced 2025-08-16 07:56:06 +00:00
Ricardo Montañana
e95bd9697a
Added check_estimator in notebook test2 Added a Stree test with check_estimator
40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
import setuptools
|
|
|
|
__version__ = "0.9rc3"
|
|
__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.7',
|
|
'Natural Language :: English',
|
|
'Topic :: Scientific/Engineering :: Artificial Intelligence',
|
|
'Intended Audience :: Science/Research'
|
|
],
|
|
install_requires=[
|
|
'scikit-learn>=0.23.0',
|
|
'numpy',
|
|
'matplotlib',
|
|
'ipympl'
|
|
],
|
|
test_suite="stree.tests",
|
|
zip_safe=False
|
|
)
|