mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-31 21:02:33 +00:00
efd85b4c86
A new documentation website. And: * add actions for doc * update docstr * installation instructions for doc dev * unify README and Getting Started * rename notebook * doc about best_model_for_estimator #340 * docstr for keep_search_state #340 * DNN Co-authored-by: Qingyun Wu <qingyun.wu@psu.edu> Co-authored-by: Z.sk <shaokunzhang@psu.edu>
15 lines
363 B
Python
15 lines
363 B
Python
from sklearn.datasets import fetch_openml
|
|
from flaml import AutoML
|
|
|
|
X_train, y_train = fetch_openml(name="credit-g", return_X_y=True, as_frame=False)
|
|
# not a real learning to rank dataaset
|
|
groups = [200] * 4 + [100] * 2 # group counts
|
|
automl = AutoML()
|
|
automl.fit(
|
|
X_train,
|
|
y_train,
|
|
groups=groups,
|
|
task="rank",
|
|
time_budget=1, # in seconds
|
|
)
|