mirror of
https://github.com/microsoft/autogen.git
synced 2025-08-31 12:00:11 +00:00
16 lines
363 B
Python
16 lines
363 B
Python
from flaml import AutoML
|
|
|
|
|
|
def _test_ray_classification():
|
|
from sklearn.datasets import make_classification
|
|
import ray
|
|
|
|
ray.init(address="auto")
|
|
X, y = make_classification(1000, 10)
|
|
automl = AutoML()
|
|
automl.fit(X, y, time_budget=10, task="classification", n_concurrent_trials=2)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
_test_ray_classification()
|