forked from 4paradigm/AutoX
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
28 lines (27 loc) · 949 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
# -*- coding: utf-8 -*-
from distutils.core import setup
from setuptools import find_packages
setup(name="autox",
version="0.1.0",
description="automl for competition",
author="caihengxing, zhongrunxing",
author_email="caihengxing@4paradigm.com",
install_requires=[
'lightgbm',
'xgboost',
# 'pytorch-tabnet',
'torch',
'numpy',
'pandas',
'sklearn',
'tqdm',
'optuna'
],
python_requires='>=3.6',
# packages=[],
packages=find_packages(exclude=['data','demo','sub','test','README.md','run_oneclick.py']),
#ext_modules=cythonize("cli_examples/primes.pyx"), #Cython extension demo
#package_data={"cli_examples": ["prmies.pyx"]}, #force sdist to keep the .pyx files
include_package_data=True,
zip_safe=False # not install as zip file
)