mirror of
https://github.com/microsoft/autogen.git
synced 2025-10-31 09:50:11 +00:00
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
|
from flaml.data import load_openml_dataset
|
||
|
|
from flaml.ml import ExtraTreesEstimator
|
||
|
|
from flaml import AutoML
|
||
|
|
|
||
|
|
X_train, X_test, y_train, y_test = load_openml_dataset(dataset_id=1169, data_dir="./")
|
||
|
|
X_train = X_train.iloc[:1000]
|
||
|
|
y_train = y_train.iloc[:1000]
|
||
|
|
|
||
|
|
|
||
|
|
class ExtraTreesEstimatorSeeded(ExtraTreesEstimator):
|
||
|
|
"""ExtraTreesEstimator for reproducible FLAML run."""
|
||
|
|
|
||
|
|
def config2params(self, config: dict) -> dict:
|
||
|
|
params = super().config2params(config)
|
||
|
|
params["random_state"] = 0
|
||
|
|
return params
|
||
|
|
|
||
|
|
|
||
|
|
settings = {
|
||
|
|
"time_budget": 1e10, # total running time in seconds
|
||
|
|
"max_iter": 3,
|
||
|
|
"metric": "ap", # average_precision
|
||
|
|
"task": "classification", # task type
|
||
|
|
"seed": 7654321, # random seed
|
||
|
|
"estimator_list": ["extra_trees_seeded"],
|
||
|
|
"verbose": False,
|
||
|
|
}
|
||
|
|
|
||
|
|
for trial_num in range(8):
|
||
|
|
automl = AutoML()
|
||
|
|
automl.add_learner(
|
||
|
|
learner_name="extra_trees_seeded", learner_class=ExtraTreesEstimatorSeeded
|
||
|
|
)
|
||
|
|
automl.fit(X_train=X_train, y_train=y_train, **settings)
|
||
|
|
print(automl.best_loss)
|
||
|
|
print(automl.best_config)
|