mirror of
https://github.com/microsoft/autogen.git
synced 2025-09-08 07:46:24 +00:00
ca35fa969f
* refactoring TransformersEstimator to support default and custom_hp * handling starting_points not in search space * addressing starting point more than max_iter * fixing upper < lower bug
27 lines
600 B
Python
27 lines
600 B
Python
from utils import get_toy_data_multiclassclassification, get_automl_settings
|
|
|
|
|
|
def test_classification_head():
|
|
from flaml import AutoML
|
|
import requests
|
|
|
|
X_train, y_train, X_val, y_val = get_toy_data_multiclassclassification()
|
|
automl = AutoML()
|
|
|
|
automl_settings = get_automl_settings()
|
|
|
|
try:
|
|
automl.fit(
|
|
X_train=X_train,
|
|
y_train=y_train,
|
|
X_val=X_val,
|
|
y_val=y_val,
|
|
**automl_settings
|
|
)
|
|
except requests.exceptions.HTTPError:
|
|
return
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_classification_head()
|