mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-12 03:21:14 +00:00
0d3a0bfab6
* pickle the AutoML object * get best model per estimator * test deberta * stateless API * pickle the AutoML object * get best model per estimator * test deberta * stateless API * prevent divide by zero * test roberta * BlendSearchTuner * sync * version number * update gitignore * delta time * reindex columns when dropping int-indexed columns * add seed * add seed in Args * merge * init upload of ChaCha * remove redundancy * add back catboost * improve AutoVW API * set min_resource_lease in VWOnlineTrial * docstr * rename * docstr * add docstr * improve API and documentation * fix name * docstr * naming * remove max_resource in scheduler * add TODO in flow2 * remove redundancy in rearcher * add input type * adapt code from ray.tune * move files * naming * documentation * fix import error * fix format issues * remove cb in worse than test * improve _generate_all_comb * remove ray tune * naming * VowpalWabbitTrial * import error * import error * merge test code * scheduler import * fix import * remove * import, minor bug and version * Float or Categorical * fix default * add test_autovw.py * add vowpalwabbit and openml * lint * reorg * lint * indent * add autovw notebook * update notebook * update log msg and autovw notebook * update autovw notebook * update autovw notebook * add available strings for model_select_policy * string for metric * Update vw format in flaml/onlineml/trial.py Co-authored-by: olgavrou <olgavrou@gmail.com> * make init_config optional * add _setup_trial_runner and update notebook * space Co-authored-by: Chi Wang (MSR) <chiw@microsoft.com> Co-authored-by: Chi Wang <wang.chi@microsoft.com> Co-authored-by: Qingyun Wu <qiw@microsoft.com> Co-authored-by: olgavrou <olgavrou@gmail.com>
79 lines
1.8 KiB
Python
79 lines
1.8 KiB
Python
import setuptools
|
|
import os
|
|
|
|
here = os.path.abspath(os.path.dirname(__file__))
|
|
|
|
with open("README.md", "r") as fh:
|
|
long_description = fh.read()
|
|
|
|
|
|
# Get the code version
|
|
version = {}
|
|
with open(os.path.join(here, "flaml/version.py")) as fp:
|
|
exec(fp.read(), version)
|
|
__version__ = version["__version__"]
|
|
|
|
install_requires = [
|
|
"NumPy>=1.16.2",
|
|
"lightgbm>=2.3.1",
|
|
"xgboost>=0.90",
|
|
"scipy>=1.4.1",
|
|
"catboost>=0.23",
|
|
"scikit-learn>=0.23.2",
|
|
],
|
|
|
|
|
|
setuptools.setup(
|
|
name="FLAML",
|
|
version=__version__,
|
|
author="Microsoft Corporation",
|
|
author_email="hpo@microsoft.com",
|
|
description="A fast and lightweight autoML system",
|
|
long_description=long_description,
|
|
long_description_content_type="text/markdown",
|
|
url="https://github.com/microsoft/FLAML",
|
|
packages=setuptools.find_packages(),
|
|
install_requires=install_requires,
|
|
extras_require={
|
|
"notebook": [
|
|
"openml==0.10.2",
|
|
"jupyter",
|
|
"matplotlib==3.2.0",
|
|
"rgf-python",
|
|
"vowpalwabbit",
|
|
],
|
|
"test": [
|
|
"flake8>=3.8.4",
|
|
"pytest>=6.1.1",
|
|
"coverage>=5.3",
|
|
"xgboost<1.3",
|
|
"rgf-python",
|
|
"optuna==2.3.0",
|
|
"vowpalwabbit",
|
|
"openml",
|
|
],
|
|
"blendsearch": [
|
|
"optuna==2.3.0"
|
|
],
|
|
"ray": [
|
|
"ray[tune]==1.2.0",
|
|
"pyyaml<5.3.1",
|
|
],
|
|
"azureml": [
|
|
"azureml-mlflow",
|
|
],
|
|
"nni": [
|
|
"nni",
|
|
],
|
|
"vw": [
|
|
"vowpalwabbit",
|
|
]
|
|
},
|
|
classifiers=[
|
|
"Programming Language :: Python :: 3",
|
|
"License :: OSI Approved :: MIT License",
|
|
"Operating System :: OS Independent",
|
|
],
|
|
python_requires=">=3.6",
|
|
)
|