mirror of
https://github.com/Doctorado-ML/benchmark.git
synced 2025-08-15 07:25:53 +00:00
78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
import setuptools
|
|
import os
|
|
|
|
|
|
def readme():
|
|
with open("README.md") as f:
|
|
return f.read()
|
|
|
|
|
|
def get_data(field):
|
|
item = ""
|
|
file_name = "_version.py" if field == "version" else "__init__.py"
|
|
with open(os.path.join("benchmark", file_name)) as f:
|
|
for line in f.readlines():
|
|
if line.startswith(f"__{field}__"):
|
|
delim = '"' if '"' in line else "'"
|
|
item = line.split(delim)[1]
|
|
break
|
|
else:
|
|
raise RuntimeError(f"Unable to find {field} string.")
|
|
return item
|
|
|
|
|
|
def import_scripts():
|
|
result = []
|
|
names = os.listdir(os.path.join("benchmark", "scripts"))
|
|
for name in names:
|
|
result.append(os.path.join("benchmark", "scripts", name))
|
|
return result
|
|
|
|
|
|
setuptools.setup(
|
|
name="benchmark",
|
|
version=get_data("version"),
|
|
license=get_data("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/benchmark",
|
|
author=get_data("author"),
|
|
author_email=get_data("author_email"),
|
|
keywords="scikit-learn oblique-classifier oblique-decision-tree decision-\
|
|
tree svm svc",
|
|
classifiers=[
|
|
"Development Status :: 4 - Beta",
|
|
"License :: OSI Approved :: " + get_data("license"),
|
|
"Programming Language :: Python :: 3.8",
|
|
"Natural Language :: English",
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
|
"Intended Audience :: Science/Research",
|
|
],
|
|
install_requires=[
|
|
"scikit-learn",
|
|
"odte",
|
|
"pandas",
|
|
"mufs",
|
|
"xlsxwriter",
|
|
"tqdm",
|
|
],
|
|
zip_safe=False,
|
|
entry_points={
|
|
"console_scripts": [
|
|
"be_list=benchmark.scripts.be_list:main",
|
|
"be_report=benchmark.scripts.be_report:main",
|
|
"be_main=benchmark.scripts.be_main:main",
|
|
"be_benchmark=benchmark.scripts.be_benchmark:main",
|
|
"be_best=benchmark.scripts.be_best:main",
|
|
"be_build_best=benchmark.scripts.be_build_best:main",
|
|
"be_grid=benchmark.scripts.be_grid:main",
|
|
"be_pair_check=benchmark.scripts.be_pair_check:main",
|
|
"be_print_strees=benchmark.scripts.be_print_strees:main",
|
|
"be_repara=benchmark.scripts.be_repara:main",
|
|
"be_summary=benchmark.scripts.be_summary:main",
|
|
],
|
|
},
|
|
)
|