FLAML 2.1.2__tar.gz → 2.2.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (119) hide show
  1. {FLAML-2.1.2 → flaml-2.2.0}/FLAML.egg-info/PKG-INFO +40 -26
  2. {FLAML-2.1.2 → flaml-2.2.0}/FLAML.egg-info/requires.txt +23 -10
  3. flaml-2.2.0/NOTICE.md +295 -0
  4. {FLAML-2.1.2 → flaml-2.2.0}/PKG-INFO +40 -26
  5. {FLAML-2.1.2 → flaml-2.2.0}/README.md +24 -15
  6. {FLAML-2.1.2 → flaml-2.2.0}/flaml/autogen/agentchat/agent.py +2 -2
  7. {FLAML-2.1.2 → flaml-2.2.0}/flaml/autogen/agentchat/assistant_agent.py +18 -18
  8. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/automl.py +7 -7
  9. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/ml.py +13 -8
  10. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/model.py +12 -6
  11. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/time_series/ts_data.py +7 -5
  12. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/time_series/ts_model.py +2 -4
  13. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/searcher/online_searcher.py +2 -2
  14. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/searcher/suggestion.py +12 -4
  15. flaml-2.2.0/flaml/version.py +1 -0
  16. {FLAML-2.1.2 → flaml-2.2.0}/setup.py +17 -10
  17. {FLAML-2.1.2 → flaml-2.2.0}/test/test_autovw.py +10 -7
  18. FLAML-2.1.2/NOTICE.md +0 -290
  19. FLAML-2.1.2/flaml/version.py +0 -1
  20. {FLAML-2.1.2 → flaml-2.2.0}/FLAML.egg-info/SOURCES.txt +0 -0
  21. {FLAML-2.1.2 → flaml-2.2.0}/FLAML.egg-info/dependency_links.txt +0 -0
  22. {FLAML-2.1.2 → flaml-2.2.0}/FLAML.egg-info/top_level.txt +0 -0
  23. {FLAML-2.1.2 → flaml-2.2.0}/LICENSE +0 -0
  24. {FLAML-2.1.2 → flaml-2.2.0}/flaml/__init__.py +0 -0
  25. {FLAML-2.1.2 → flaml-2.2.0}/flaml/autogen/__init__.py +0 -0
  26. {FLAML-2.1.2 → flaml-2.2.0}/flaml/autogen/agentchat/__init__.py +0 -0
  27. {FLAML-2.1.2 → flaml-2.2.0}/flaml/autogen/agentchat/conversable_agent.py +0 -0
  28. {FLAML-2.1.2 → flaml-2.2.0}/flaml/autogen/agentchat/groupchat.py +0 -0
  29. {FLAML-2.1.2 → flaml-2.2.0}/flaml/autogen/agentchat/user_proxy_agent.py +0 -0
  30. {FLAML-2.1.2 → flaml-2.2.0}/flaml/autogen/code_utils.py +0 -0
  31. {FLAML-2.1.2 → flaml-2.2.0}/flaml/autogen/extensions/__init__.py +0 -0
  32. {FLAML-2.1.2 → flaml-2.2.0}/flaml/autogen/math_utils.py +0 -0
  33. {FLAML-2.1.2 → flaml-2.2.0}/flaml/autogen/oai/__init__.py +0 -0
  34. {FLAML-2.1.2 → flaml-2.2.0}/flaml/autogen/oai/completion.py +0 -0
  35. {FLAML-2.1.2 → flaml-2.2.0}/flaml/autogen/oai/openai_utils.py +0 -0
  36. {FLAML-2.1.2 → flaml-2.2.0}/flaml/autogen/retrieve_utils.py +0 -0
  37. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/__init__.py +0 -0
  38. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/contrib/__init__.py +0 -0
  39. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/contrib/histgb.py +0 -0
  40. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/data.py +0 -0
  41. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/logger.py +0 -0
  42. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/nlp/__init__.py +0 -0
  43. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/nlp/huggingface/__init__.py +0 -0
  44. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/nlp/huggingface/data_collator.py +0 -0
  45. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/nlp/huggingface/trainer.py +0 -0
  46. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/nlp/huggingface/training_args.py +0 -0
  47. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/nlp/huggingface/utils.py +0 -0
  48. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/nlp/utils.py +0 -0
  49. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/spark/__init__.py +0 -0
  50. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/spark/configs.py +0 -0
  51. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/spark/metrics.py +0 -0
  52. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/spark/utils.py +0 -0
  53. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/state.py +0 -0
  54. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/task/__init__.py +0 -0
  55. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/task/factory.py +0 -0
  56. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/task/generic_task.py +0 -0
  57. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/task/task.py +0 -0
  58. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/task/time_series_task.py +0 -0
  59. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/time_series/__init__.py +0 -0
  60. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/time_series/feature.py +0 -0
  61. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/time_series/sklearn.py +0 -0
  62. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/time_series/tft.py +0 -0
  63. {FLAML-2.1.2 → flaml-2.2.0}/flaml/automl/training_log.py +0 -0
  64. {FLAML-2.1.2 → flaml-2.2.0}/flaml/config.py +0 -0
  65. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/__init__.py +0 -0
  66. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/all/binary.json +0 -0
  67. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/all/multiclass.json +0 -0
  68. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/all/regression.json +0 -0
  69. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/estimator.py +0 -0
  70. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/extra_tree/binary.json +0 -0
  71. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/extra_tree/multiclass.json +0 -0
  72. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/extra_tree/regression.json +0 -0
  73. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/greedy.py +0 -0
  74. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/lgbm/binary.json +0 -0
  75. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/lgbm/multiclass.json +0 -0
  76. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/lgbm/regression.json +0 -0
  77. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/portfolio.py +0 -0
  78. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/regret.py +0 -0
  79. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/rf/binary.json +0 -0
  80. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/rf/multiclass.json +0 -0
  81. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/rf/regression.json +0 -0
  82. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/suggest.py +0 -0
  83. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/xgb_limitdepth/binary.json +0 -0
  84. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/xgb_limitdepth/multiclass.json +0 -0
  85. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/xgb_limitdepth/regression.json +0 -0
  86. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/xgboost/binary.json +0 -0
  87. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/xgboost/multiclass.json +0 -0
  88. {FLAML-2.1.2 → flaml-2.2.0}/flaml/default/xgboost/regression.json +0 -0
  89. {FLAML-2.1.2 → flaml-2.2.0}/flaml/ml.py +0 -0
  90. {FLAML-2.1.2 → flaml-2.2.0}/flaml/onlineml/__init__.py +0 -0
  91. {FLAML-2.1.2 → flaml-2.2.0}/flaml/onlineml/autovw.py +0 -0
  92. {FLAML-2.1.2 → flaml-2.2.0}/flaml/onlineml/trial.py +0 -0
  93. {FLAML-2.1.2 → flaml-2.2.0}/flaml/onlineml/trial_runner.py +0 -0
  94. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/__init__.py +0 -0
  95. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/analysis.py +0 -0
  96. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/result.py +0 -0
  97. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/sample.py +0 -0
  98. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/scheduler/__init__.py +0 -0
  99. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/scheduler/online_scheduler.py +0 -0
  100. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/scheduler/trial_scheduler.py +0 -0
  101. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/searcher/__init__.py +0 -0
  102. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/searcher/blendsearch.py +0 -0
  103. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/searcher/cfo_cat.py +0 -0
  104. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/searcher/flow2.py +0 -0
  105. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/searcher/search_thread.py +0 -0
  106. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/searcher/variant_generator.py +0 -0
  107. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/space.py +0 -0
  108. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/spark/__init__.py +0 -0
  109. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/spark/utils.py +0 -0
  110. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/trial.py +0 -0
  111. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/trial_runner.py +0 -0
  112. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/tune.py +0 -0
  113. {FLAML-2.1.2 → flaml-2.2.0}/flaml/tune/utils.py +0 -0
  114. {FLAML-2.1.2 → flaml-2.2.0}/pyproject.toml +0 -0
  115. {FLAML-2.1.2 → flaml-2.2.0}/setup.cfg +0 -0
  116. {FLAML-2.1.2 → flaml-2.2.0}/test/test_conda_distribution.py +0 -0
  117. {FLAML-2.1.2 → flaml-2.2.0}/test/test_gpu.py +0 -0
  118. {FLAML-2.1.2 → flaml-2.2.0}/test/test_model.py +0 -0
  119. {FLAML-2.1.2 → flaml-2.2.0}/test/test_version.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: FLAML
3
- Version: 2.1.2
3
+ Version: 2.2.0
4
4
  Summary: A fast library for automated machine learning and tuning
5
5
  Home-page: https://github.com/microsoft/FLAML
6
6
  Author: Microsoft Corporation
@@ -15,30 +15,33 @@ License-File: NOTICE.md
15
15
  Requires-Dist: NumPy>=1.17
16
16
  Provides-Extra: automl
17
17
  Requires-Dist: lightgbm>=2.3.1; extra == "automl"
18
- Requires-Dist: xgboost>=0.90; extra == "automl"
18
+ Requires-Dist: xgboost<3.0.0,>=0.90; extra == "automl"
19
19
  Requires-Dist: scipy>=1.4.1; extra == "automl"
20
20
  Requires-Dist: pandas>=1.1.4; extra == "automl"
21
- Requires-Dist: scikit-learn>=0.24; extra == "automl"
21
+ Requires-Dist: scikit-learn>=1.0.0; extra == "automl"
22
22
  Provides-Extra: notebook
23
23
  Requires-Dist: jupyter; extra == "notebook"
24
24
  Provides-Extra: spark
25
25
  Requires-Dist: pyspark>=3.2.0; extra == "spark"
26
26
  Requires-Dist: joblibspark>=0.5.0; extra == "spark"
27
+ Requires-Dist: joblib<=1.3.2; extra == "spark"
27
28
  Provides-Extra: test
29
+ Requires-Dist: jupyter; extra == "test"
28
30
  Requires-Dist: lightgbm>=2.3.1; extra == "test"
29
- Requires-Dist: xgboost>=0.90; extra == "test"
31
+ Requires-Dist: xgboost<2.0.0,>=0.90; extra == "test"
30
32
  Requires-Dist: scipy>=1.4.1; extra == "test"
31
33
  Requires-Dist: pandas>=1.1.4; extra == "test"
32
- Requires-Dist: scikit-learn>=0.24; extra == "test"
34
+ Requires-Dist: scikit-learn>=1.0.0; extra == "test"
33
35
  Requires-Dist: thop; extra == "test"
34
36
  Requires-Dist: pytest>=6.1.1; extra == "test"
35
37
  Requires-Dist: coverage>=5.3; extra == "test"
36
38
  Requires-Dist: pre-commit; extra == "test"
37
39
  Requires-Dist: torch; extra == "test"
38
40
  Requires-Dist: torchvision; extra == "test"
39
- Requires-Dist: catboost<1.2,>=0.26; extra == "test"
41
+ Requires-Dist: catboost<1.2,>=0.26; python_version < "3.11" and extra == "test"
42
+ Requires-Dist: catboost>=0.26; python_version >= "3.11" and extra == "test"
40
43
  Requires-Dist: rgf-python; extra == "test"
41
- Requires-Dist: optuna==2.8.0; extra == "test"
44
+ Requires-Dist: optuna<=3.6.1,>=2.8.0; extra == "test"
42
45
  Requires-Dist: openml; extra == "test"
43
46
  Requires-Dist: statsmodels>=0.12.2; extra == "test"
44
47
  Requires-Dist: psutil==5.8.0; extra == "test"
@@ -49,10 +52,11 @@ Requires-Dist: nltk; extra == "test"
49
52
  Requires-Dist: rouge_score; extra == "test"
50
53
  Requires-Dist: hcrystalball==0.1.10; extra == "test"
51
54
  Requires-Dist: seqeval; extra == "test"
52
- Requires-Dist: pytorch-forecasting<=0.10.1,>=0.9.0; extra == "test"
55
+ Requires-Dist: pytorch-forecasting<=0.10.1,>=0.9.0; python_version < "3.11" and extra == "test"
53
56
  Requires-Dist: mlflow; extra == "test"
54
57
  Requires-Dist: pyspark>=3.2.0; extra == "test"
55
58
  Requires-Dist: joblibspark>=0.5.0; extra == "test"
59
+ Requires-Dist: joblib<=1.3.2; extra == "test"
56
60
  Requires-Dist: nbconvert; extra == "test"
57
61
  Requires-Dist: nbformat; extra == "test"
58
62
  Requires-Dist: ipykernel; extra == "test"
@@ -64,9 +68,10 @@ Requires-Dist: pydantic==1.10.9; extra == "test"
64
68
  Requires-Dist: sympy; extra == "test"
65
69
  Requires-Dist: wolframalpha; extra == "test"
66
70
  Provides-Extra: catboost
67
- Requires-Dist: catboost>=0.26; extra == "catboost"
71
+ Requires-Dist: catboost<1.2,>=0.26; python_version < "3.11" and extra == "catboost"
72
+ Requires-Dist: catboost<=1.2.5,>=0.26; python_version >= "3.11" and extra == "catboost"
68
73
  Provides-Extra: blendsearch
69
- Requires-Dist: optuna==2.8.0; extra == "blendsearch"
74
+ Requires-Dist: optuna<=3.6.1,>=2.8.0; extra == "blendsearch"
70
75
  Requires-Dist: packaging; extra == "blendsearch"
71
76
  Provides-Extra: ray
72
77
  Requires-Dist: ray[tune]~=1.13; extra == "ray"
@@ -130,7 +135,7 @@ Requires-Dist: tiktoken; extra == "retrievechat"
130
135
  Requires-Dist: sentence_transformers; extra == "retrievechat"
131
136
  Provides-Extra: synapse
132
137
  Requires-Dist: joblibspark>=0.5.0; extra == "synapse"
133
- Requires-Dist: optuna==2.8.0; extra == "synapse"
138
+ Requires-Dist: optuna<=3.6.1,>=2.8.0; extra == "synapse"
134
139
  Requires-Dist: pyspark>=3.2.0; extra == "synapse"
135
140
  Provides-Extra: autozero
136
141
  Requires-Dist: scikit-learn; extra == "autozero"
@@ -143,8 +148,8 @@ Requires-Dist: packaging; extra == "autozero"
143
148
  ![Python Version](https://img.shields.io/badge/3.8%20%7C%203.9%20%7C%203.10-blue)
144
149
  [![Downloads](https://pepy.tech/badge/flaml)](https://pepy.tech/project/flaml)
145
150
  [![](https://img.shields.io/discord/1025786666260111483?logo=discord&style=flat)](https://discord.gg/Cppx2vSPVP)
146
- <!-- [![Join the chat at https://gitter.im/FLAMLer/community](https://badges.gitter.im/FLAMLer/community.svg)](https://gitter.im/FLAMLer/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge) -->
147
151
 
152
+ <!-- [![Join the chat at https://gitter.im/FLAMLer/community](https://badges.gitter.im/FLAMLer/community.svg)](https://gitter.im/FLAMLer/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge) -->
148
153
 
149
154
  # A Fast Library for Automated Machine Learning & Tuning
150
155
 
@@ -163,15 +168,15 @@ Requires-Dist: packaging; extra == "autozero"
163
168
 
164
169
  :fire: FLAML supports Code-First AutoML & Tuning – Private Preview in [Microsoft Fabric Data Science](https://learn.microsoft.com/en-us/fabric/data-science/).
165
170
 
166
-
167
171
  ## What is FLAML
172
+
168
173
  FLAML is a lightweight Python library for efficient automation of machine
169
174
  learning and AI operations. It automates workflow based on large language models, machine learning models, etc.
170
175
  and optimizes their performance.
171
176
 
172
- * FLAML enables building next-gen GPT-X applications based on multi-agent conversations with minimal effort. It simplifies the orchestration, automation and optimization of a complex GPT-X workflow. It maximizes the performance of GPT-X models and augments their weakness.
173
- * For common machine learning tasks like classification and regression, it quickly finds quality models for user-provided data with low computational resources. It is easy to customize or extend. Users can find their desired customizability from a smooth range.
174
- * It supports fast and economical automatic tuning (e.g., inference hyperparameters for foundation models, configurations in MLOps/LMOps workflows, pipelines, mathematical/statistical models, algorithms, computing experiments, software configurations), capable of handling large search space with heterogeneous evaluation cost and complex constraints/guidance/early stopping.
177
+ - FLAML enables building next-gen GPT-X applications based on multi-agent conversations with minimal effort. It simplifies the orchestration, automation and optimization of a complex GPT-X workflow. It maximizes the performance of GPT-X models and augments their weakness.
178
+ - For common machine learning tasks like classification and regression, it quickly finds quality models for user-provided data with low computational resources. It is easy to customize or extend. Users can find their desired customizability from a smooth range.
179
+ - It supports fast and economical automatic tuning (e.g., inference hyperparameters for foundation models, configurations in MLOps/LMOps workflows, pipelines, mathematical/statistical models, algorithms, computing experiments, software configurations), capable of handling large search space with heterogeneous evaluation cost and complex constraints/guidance/early stopping.
175
180
 
176
181
  FLAML is powered by a series of [research studies](https://microsoft.github.io/FLAML/docs/Research/) from Microsoft Research and collaborators such as Penn State University, Stevens Institute of Technology, University of Washington, and University of Waterloo.
177
182
 
@@ -186,6 +191,7 @@ pip install flaml
186
191
  ```
187
192
 
188
193
  Minimal dependencies are installed without extra options. You can install extra options based on the feature you need. For example, use the following to install the dependencies needed by the [`autogen`](https://microsoft.github.io/autogen/) package.
194
+
189
195
  ```bash
190
196
  pip install "flaml[autogen]"
191
197
  ```
@@ -195,18 +201,24 @@ Each of the [`notebook examples`](https://github.com/microsoft/FLAML/tree/main/n
195
201
 
196
202
  ## Quickstart
197
203
 
198
- * (New) The [autogen](https://microsoft.github.io/autogen/) package enables the next-gen GPT-X applications with a generic multi-agent conversation framework.
199
- It offers customizable and conversable agents which integrate LLMs, tools and human.
200
- By automating chat among multiple capable agents, one can easily make them collectively perform tasks autonomously or with human feedback, including tasks that require using tools via code. For example,
204
+ - (New) The [autogen](https://microsoft.github.io/autogen/) package enables the next-gen GPT-X applications with a generic multi-agent conversation framework.
205
+ It offers customizable and conversable agents which integrate LLMs, tools and human.
206
+ By automating chat among multiple capable agents, one can easily make them collectively perform tasks autonomously or with human feedback, including tasks that require using tools via code. For example,
207
+
201
208
  ```python
202
209
  from flaml import autogen
210
+
203
211
  assistant = autogen.AssistantAgent("assistant")
204
212
  user_proxy = autogen.UserProxyAgent("user_proxy")
205
- user_proxy.initiate_chat(assistant, message="Show me the YTD gain of 10 largest technology companies as of today.")
213
+ user_proxy.initiate_chat(
214
+ assistant,
215
+ message="Show me the YTD gain of 10 largest technology companies as of today.",
216
+ )
206
217
  # This initiates an automated chat between the two agents to solve the task
207
218
  ```
208
219
 
209
220
  Autogen also helps maximize the utility out of the expensive LLMs such as ChatGPT and GPT-4. It offers a drop-in replacement of `openai.Completion` or `openai.ChatCompletion` with powerful functionalites like tuning, caching, templating, filtering. For example, you can optimize generations by LLM with your own tuning data, success metrics and budgets.
221
+
210
222
  ```python
211
223
  # perform tuning
212
224
  config, analysis = autogen.Completion.tune(
@@ -221,30 +233,32 @@ config, analysis = autogen.Completion.tune(
221
233
  # perform inference for a test instance
222
234
  response = autogen.Completion.create(context=test_instance, **config)
223
235
  ```
224
- * With three lines of code, you can start using this economical and fast
225
- AutoML engine as a [scikit-learn style estimator](https://microsoft.github.io/FLAML/docs/Use-Cases/Task-Oriented-AutoML).
236
+
237
+ - With three lines of code, you can start using this economical and fast
238
+ AutoML engine as a [scikit-learn style estimator](https://microsoft.github.io/FLAML/docs/Use-Cases/Task-Oriented-AutoML).
226
239
 
227
240
  ```python
228
241
  from flaml import AutoML
242
+
229
243
  automl = AutoML()
230
244
  automl.fit(X_train, y_train, task="classification")
231
245
  ```
232
246
 
233
- * You can restrict the learners and use FLAML as a fast hyperparameter tuning
234
- tool for XGBoost, LightGBM, Random Forest etc. or a [customized learner](https://microsoft.github.io/FLAML/docs/Use-Cases/Task-Oriented-AutoML#estimator-and-search-space).
247
+ - You can restrict the learners and use FLAML as a fast hyperparameter tuning
248
+ tool for XGBoost, LightGBM, Random Forest etc. or a [customized learner](https://microsoft.github.io/FLAML/docs/Use-Cases/Task-Oriented-AutoML#estimator-and-search-space).
235
249
 
236
250
  ```python
237
251
  automl.fit(X_train, y_train, task="classification", estimator_list=["lgbm"])
238
252
  ```
239
253
 
240
- * You can also run generic hyperparameter tuning for a [custom function](https://microsoft.github.io/FLAML/docs/Use-Cases/Tune-User-Defined-Function).
254
+ - You can also run generic hyperparameter tuning for a [custom function](https://microsoft.github.io/FLAML/docs/Use-Cases/Tune-User-Defined-Function).
241
255
 
242
256
  ```python
243
257
  from flaml import tune
244
258
  tune.run(evaluation_function, config={…}, low_cost_partial_config={…}, time_budget_s=3600)
245
259
  ```
246
260
 
247
- * [Zero-shot AutoML](https://microsoft.github.io/FLAML/docs/Use-Cases/Zero-Shot-AutoML) allows using the existing training API from lightgbm, xgboost etc. while getting the benefit of AutoML in choosing high-performance hyperparameter configurations per task.
261
+ - [Zero-shot AutoML](https://microsoft.github.io/FLAML/docs/Use-Cases/Zero-Shot-AutoML) allows using the existing training API from lightgbm, xgboost etc. while getting the benefit of AutoML in choosing high-performance hyperparameter configurations per task.
248
262
 
249
263
  ```python
250
264
  from flaml.default import LGBMRegressor
@@ -7,10 +7,10 @@ termcolor
7
7
 
8
8
  [automl]
9
9
  lightgbm>=2.3.1
10
- xgboost>=0.90
10
+ xgboost<3.0.0,>=0.90
11
11
  scipy>=1.4.1
12
12
  pandas>=1.1.4
13
- scikit-learn>=0.24
13
+ scikit-learn>=1.0.0
14
14
 
15
15
  [autozero]
16
16
  scikit-learn
@@ -27,11 +27,16 @@ xgboost==1.3.3
27
27
  pandas==1.1.4
28
28
 
29
29
  [blendsearch]
30
- optuna==2.8.0
30
+ optuna<=3.6.1,>=2.8.0
31
31
  packaging
32
32
 
33
33
  [catboost]
34
- catboost>=0.26
34
+
35
+ [catboost:python_version < "3.11"]
36
+ catboost<1.2,>=0.26
37
+
38
+ [catboost:python_version >= "3.11"]
39
+ catboost<=1.2.5,>=0.26
35
40
 
36
41
  [forecast]
37
42
  holidays<0.14
@@ -88,27 +93,28 @@ sentence_transformers
88
93
  [spark]
89
94
  pyspark>=3.2.0
90
95
  joblibspark>=0.5.0
96
+ joblib<=1.3.2
91
97
 
92
98
  [synapse]
93
99
  joblibspark>=0.5.0
94
- optuna==2.8.0
100
+ optuna<=3.6.1,>=2.8.0
95
101
  pyspark>=3.2.0
96
102
 
97
103
  [test]
104
+ jupyter
98
105
  lightgbm>=2.3.1
99
- xgboost>=0.90
106
+ xgboost<2.0.0,>=0.90
100
107
  scipy>=1.4.1
101
108
  pandas>=1.1.4
102
- scikit-learn>=0.24
109
+ scikit-learn>=1.0.0
103
110
  thop
104
111
  pytest>=6.1.1
105
112
  coverage>=5.3
106
113
  pre-commit
107
114
  torch
108
115
  torchvision
109
- catboost<1.2,>=0.26
110
116
  rgf-python
111
- optuna==2.8.0
117
+ optuna<=3.6.1,>=2.8.0
112
118
  openml
113
119
  statsmodels>=0.12.2
114
120
  psutil==5.8.0
@@ -119,10 +125,10 @@ nltk
119
125
  rouge_score
120
126
  hcrystalball==0.1.10
121
127
  seqeval
122
- pytorch-forecasting<=0.10.1,>=0.9.0
123
128
  mlflow
124
129
  pyspark>=3.2.0
125
130
  joblibspark>=0.5.0
131
+ joblib<=1.3.2
126
132
  nbconvert
127
133
  nbformat
128
134
  ipykernel
@@ -134,6 +140,13 @@ pydantic==1.10.9
134
140
  sympy
135
141
  wolframalpha
136
142
 
143
+ [test:python_version < "3.11"]
144
+ catboost<1.2,>=0.26
145
+ pytorch-forecasting<=0.10.1,>=0.9.0
146
+
147
+ [test:python_version >= "3.11"]
148
+ catboost>=0.26
149
+
137
150
  [ts_forecast]
138
151
  holidays<0.14
139
152
  prophet>=1.0.1
flaml-2.2.0/NOTICE.md ADDED
@@ -0,0 +1,295 @@
1
+ # NOTICES
2
+
3
+ This repository incorporates material as listed below or described in the code.
4
+
5
+ ## Component. Ray.
6
+
7
+ Code in tune/\[analysis.py, sample.py, trial.py, result.py\],
8
+ searcher/\[suggestion.py, variant_generator.py\], and scheduler/trial_scheduler.py is adapted from
9
+ https://github.com/ray-project/ray/blob/master/python/ray/tune/
10
+
11
+ ## Open Source License/Copyright Notice.
12
+
13
+ Apache License
14
+ Version 2.0, January 2004
15
+ http://www.apache.org/licenses/
16
+
17
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
18
+
19
+ 1. Definitions.
20
+
21
+ "License" shall mean the terms and conditions for use, reproduction,
22
+ and distribution as defined by Sections 1 through 9 of this document.
23
+
24
+ "Licensor" shall mean the copyright owner or entity authorized by
25
+ the copyright owner that is granting the License.
26
+
27
+ "Legal Entity" shall mean the union of the acting entity and all
28
+ other entities that control, are controlled by, or are under common
29
+ control with that entity. For the purposes of this definition,
30
+ "control" means (i) the power, direct or indirect, to cause the
31
+ direction or management of such entity, whether by contract or
32
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
33
+ outstanding shares, or (iii) beneficial ownership of such entity.
34
+
35
+ "You" (or "Your") shall mean an individual or Legal Entity
36
+ exercising permissions granted by this License.
37
+
38
+ "Source" form shall mean the preferred form for making modifications,
39
+ including but not limited to software source code, documentation
40
+ source, and configuration files.
41
+
42
+ "Object" form shall mean any form resulting from mechanical
43
+ transformation or translation of a Source form, including but
44
+ not limited to compiled object code, generated documentation,
45
+ and conversions to other media types.
46
+
47
+ "Work" shall mean the work of authorship, whether in Source or
48
+ Object form, made available under the License, as indicated by a
49
+ copyright notice that is included in or attached to the work
50
+ (an example is provided in the Appendix below).
51
+
52
+ "Derivative Works" shall mean any work, whether in Source or Object
53
+ form, that is based on (or derived from) the Work and for which the
54
+ editorial revisions, annotations, elaborations, or other modifications
55
+ represent, as a whole, an original work of authorship. For the purposes
56
+ of this License, Derivative Works shall not include works that remain
57
+ separable from, or merely link (or bind by name) to the interfaces of,
58
+ the Work and Derivative Works thereof.
59
+
60
+ "Contribution" shall mean any work of authorship, including
61
+ the original version of the Work and any modifications or additions
62
+ to that Work or Derivative Works thereof, that is intentionally
63
+ submitted to Licensor for inclusion in the Work by the copyright owner
64
+ or by an individual or Legal Entity authorized to submit on behalf of
65
+ the copyright owner. For the purposes of this definition, "submitted"
66
+ means any form of electronic, verbal, or written communication sent
67
+ to the Licensor or its representatives, including but not limited to
68
+ communication on electronic mailing lists, source code control systems,
69
+ and issue tracking systems that are managed by, or on behalf of, the
70
+ Licensor for the purpose of discussing and improving the Work, but
71
+ excluding communication that is conspicuously marked or otherwise
72
+ designated in writing by the copyright owner as "Not a Contribution."
73
+
74
+ "Contributor" shall mean Licensor and any individual or Legal Entity
75
+ on behalf of whom a Contribution has been received by Licensor and
76
+ subsequently incorporated within the Work.
77
+
78
+ 1. Grant of Copyright License. Subject to the terms and conditions of
79
+ this License, each Contributor hereby grants to You a perpetual,
80
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
81
+ copyright license to reproduce, prepare Derivative Works of,
82
+ publicly display, publicly perform, sublicense, and distribute the
83
+ Work and such Derivative Works in Source or Object form.
84
+
85
+ 1. Grant of Patent License. Subject to the terms and conditions of
86
+ this License, each Contributor hereby grants to You a perpetual,
87
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
88
+ (except as stated in this section) patent license to make, have made,
89
+ use, offer to sell, sell, import, and otherwise transfer the Work,
90
+ where such license applies only to those patent claims licensable
91
+ by such Contributor that are necessarily infringed by their
92
+ Contribution(s) alone or by combination of their Contribution(s)
93
+ with the Work to which such Contribution(s) was submitted. If You
94
+ institute patent litigation against any entity (including a
95
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
96
+ or a Contribution incorporated within the Work constitutes direct
97
+ or contributory patent infringement, then any patent licenses
98
+ granted to You under this License for that Work shall terminate
99
+ as of the date such litigation is filed.
100
+
101
+ 1. Redistribution. You may reproduce and distribute copies of the
102
+ Work or Derivative Works thereof in any medium, with or without
103
+ modifications, and in Source or Object form, provided that You
104
+ meet the following conditions:
105
+
106
+ (a) You must give any other recipients of the Work or
107
+ Derivative Works a copy of this License; and
108
+
109
+ (b) You must cause any modified files to carry prominent notices
110
+ stating that You changed the files; and
111
+
112
+ (c) You must retain, in the Source form of any Derivative Works
113
+ that You distribute, all copyright, patent, trademark, and
114
+ attribution notices from the Source form of the Work,
115
+ excluding those notices that do not pertain to any part of
116
+ the Derivative Works; and
117
+
118
+ (d) If the Work includes a "NOTICE" text file as part of its
119
+ distribution, then any Derivative Works that You distribute must
120
+ include a readable copy of the attribution notices contained
121
+ within such NOTICE file, excluding those notices that do not
122
+ pertain to any part of the Derivative Works, in at least one
123
+ of the following places: within a NOTICE text file distributed
124
+ as part of the Derivative Works; within the Source form or
125
+ documentation, if provided along with the Derivative Works; or,
126
+ within a display generated by the Derivative Works, if and
127
+ wherever such third-party notices normally appear. The contents
128
+ of the NOTICE file are for informational purposes only and
129
+ do not modify the License. You may add Your own attribution
130
+ notices within Derivative Works that You distribute, alongside
131
+ or as an addendum to the NOTICE text from the Work, provided
132
+ that such additional attribution notices cannot be construed
133
+ as modifying the License.
134
+
135
+ You may add Your own copyright statement to Your modifications and
136
+ may provide additional or different license terms and conditions
137
+ for use, reproduction, or distribution of Your modifications, or
138
+ for any such Derivative Works as a whole, provided Your use,
139
+ reproduction, and distribution of the Work otherwise complies with
140
+ the conditions stated in this License.
141
+
142
+ 1. Submission of Contributions. Unless You explicitly state otherwise,
143
+ any Contribution intentionally submitted for inclusion in the Work
144
+ by You to the Licensor shall be under the terms and conditions of
145
+ this License, without any additional terms or conditions.
146
+ Notwithstanding the above, nothing herein shall supersede or modify
147
+ the terms of any separate license agreement you may have executed
148
+ with Licensor regarding such Contributions.
149
+
150
+ 1. Trademarks. This License does not grant permission to use the trade
151
+ names, trademarks, service marks, or product names of the Licensor,
152
+ except as required for reasonable and customary use in describing the
153
+ origin of the Work and reproducing the content of the NOTICE file.
154
+
155
+ 1. Disclaimer of Warranty. Unless required by applicable law or
156
+ agreed to in writing, Licensor provides the Work (and each
157
+ Contributor provides its Contributions) on an "AS IS" BASIS,
158
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
159
+ implied, including, without limitation, any warranties or conditions
160
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
161
+ PARTICULAR PURPOSE. You are solely responsible for determining the
162
+ appropriateness of using or redistributing the Work and assume any
163
+ risks associated with Your exercise of permissions under this License.
164
+
165
+ 1. Limitation of Liability. In no event and under no legal theory,
166
+ whether in tort (including negligence), contract, or otherwise,
167
+ unless required by applicable law (such as deliberate and grossly
168
+ negligent acts) or agreed to in writing, shall any Contributor be
169
+ liable to You for damages, including any direct, indirect, special,
170
+ incidental, or consequential damages of any character arising as a
171
+ result of this License or out of the use or inability to use the
172
+ Work (including but not limited to damages for loss of goodwill,
173
+ work stoppage, computer failure or malfunction, or any and all
174
+ other commercial damages or losses), even if such Contributor
175
+ has been advised of the possibility of such damages.
176
+
177
+ 1. Accepting Warranty or Additional Liability. While redistributing
178
+ the Work or Derivative Works thereof, You may choose to offer,
179
+ and charge a fee for, acceptance of support, warranty, indemnity,
180
+ or other liability obligations and/or rights consistent with this
181
+ License. However, in accepting such obligations, You may act only
182
+ on Your own behalf and on Your sole responsibility, not on behalf
183
+ of any other Contributor, and only if You agree to indemnify,
184
+ defend, and hold each Contributor harmless for any liability
185
+ incurred by, or claims asserted against, such Contributor by reason
186
+ of your accepting any such warranty or additional liability.
187
+
188
+ END OF TERMS AND CONDITIONS
189
+
190
+ APPENDIX: How to apply the Apache License to your work.
191
+
192
+ ```
193
+ To apply the Apache License to your work, attach the following
194
+ boilerplate notice, with the fields enclosed by brackets "{}"
195
+ replaced with your own identifying information. (Don't include
196
+ the brackets!) The text should be enclosed in the appropriate
197
+ comment syntax for the file format. We also recommend that a
198
+ file or class name and description of purpose be included on the
199
+ same "printed page" as the copyright notice for easier
200
+ identification within third-party archives.
201
+ ```
202
+
203
+ Copyright {yyyy} {name of copyright owner}
204
+
205
+ Licensed under the Apache License, Version 2.0 (the "License");
206
+ you may not use this file except in compliance with the License.
207
+ You may obtain a copy of the License at
208
+
209
+ ```
210
+ http://www.apache.org/licenses/LICENSE-2.0
211
+ ```
212
+
213
+ Unless required by applicable law or agreed to in writing, software
214
+ distributed under the License is distributed on an "AS IS" BASIS,
215
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
216
+ See the License for the specific language governing permissions and
217
+ limitations under the License.
218
+
219
+ ______________________________________________________________________
220
+
221
+ Code in python/ray/rllib/{evolution_strategies, dqn} adapted from
222
+ https://github.com/openai (MIT License)
223
+
224
+ Copyright (c) 2016 OpenAI (http://openai.com)
225
+
226
+ Permission is hereby granted, free of charge, to any person obtaining a copy
227
+ of this software and associated documentation files (the "Software"), to deal
228
+ in the Software without restriction, including without limitation the rights
229
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
230
+ copies of the Software, and to permit persons to whom the Software is
231
+ furnished to do so, subject to the following conditions:
232
+
233
+ The above copyright notice and this permission notice shall be included in
234
+ all copies or substantial portions of the Software.
235
+
236
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
237
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
238
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
239
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
240
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
241
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
242
+ THE SOFTWARE.
243
+
244
+ ______________________________________________________________________
245
+
246
+ Code in python/ray/rllib/impala/vtrace.py from
247
+ https://github.com/deepmind/scalable_agent
248
+
249
+ Copyright 2018 Google LLC
250
+
251
+ Licensed under the Apache License, Version 2.0 (the "License");
252
+ you may not use this file except in compliance with the License.
253
+ You may obtain a copy of the License at
254
+
255
+ ```
256
+ https://www.apache.org/licenses/LICENSE-2.0
257
+ ```
258
+
259
+ Unless required by applicable law or agreed to in writing, software
260
+ distributed under the License is distributed on an "AS IS" BASIS,
261
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
262
+ See the License for the specific language governing permissions and
263
+ limitations under the License.
264
+
265
+ ______________________________________________________________________
266
+
267
+ Code in python/ray/rllib/ars is adapted from https://github.com/modestyachts/ARS
268
+
269
+ Copyright (c) 2018, ARS contributors (Horia Mania, Aurelia Guy, Benjamin Recht)
270
+ All rights reserved.
271
+
272
+ Redistribution and use of ARS in source and binary forms, with or without
273
+ modification, are permitted provided that the following conditions are met:
274
+
275
+ 1. Redistributions of source code must retain the above copyright notice, this
276
+ list of conditions and the following disclaimer.
277
+
278
+ 1. Redistributions in binary form must reproduce the above copyright notice,
279
+ this list of conditions and the following disclaimer in the documentation and/or
280
+ other materials provided with the distribution.
281
+
282
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
283
+ ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
284
+ WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
285
+ DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
286
+ ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
287
+ (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
288
+ LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
289
+ ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
290
+ (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
291
+ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
292
+
293
+ ______________________________________________________________________
294
+
295
+ Code in python/ray/\_private/prometheus_exporter.py is adapted from https://github.com/census-instrumentation/opencensus-python/blob/master/contrib/opencensus-ext-prometheus/opencensus/ext/prometheus/stats_exporter/__init__.py