mirror of
https://github.com/microsoft/autogen.git
synced 2025-09-18 20:53:49 +00:00
595af7a04f
* install editable package in codespace * fix test error in test_forecast * fix test error in test_space * openml version * break tests; pre-commit * skip on py10+win32 * install mlflow in test * install mlflow in [test] * skip test in windows * import * handle PermissionError * skip test in windows * skip test in windows * skip test in windows * skip test in windows * remove ts_forecast_panel from doc
95 lines
2.3 KiB
Python
95 lines
2.3 KiB
Python
import sys
|
|
import pytest
|
|
from utils import get_toy_data_seqclassification, get_automl_settings
|
|
import os
|
|
import shutil
|
|
|
|
|
|
def custom_metric(
|
|
X_test,
|
|
y_test,
|
|
estimator,
|
|
labels,
|
|
X_train,
|
|
y_train,
|
|
weight_test=None,
|
|
weight_train=None,
|
|
config=None,
|
|
groups_test=None,
|
|
groups_train=None,
|
|
):
|
|
from datasets import Dataset
|
|
from flaml.model import TransformersEstimator
|
|
|
|
if estimator._trainer is None:
|
|
trainer = estimator._init_model_for_predict()
|
|
estimator._trainer = None
|
|
else:
|
|
trainer = estimator._trainer
|
|
X_test, y_test = estimator._tokenize_text(X_test)
|
|
|
|
if y_test is not None:
|
|
eval_dataset = Dataset.from_pandas(X_test.join(y_test))
|
|
else:
|
|
eval_dataset = Dataset.from_pandas(X_test)
|
|
|
|
estimator_metric_backup = estimator._metric
|
|
estimator._metric = "rmse"
|
|
metrics = trainer.evaluate(eval_dataset)
|
|
estimator._metric = estimator_metric_backup
|
|
|
|
return metrics.pop("eval_automl_metric"), metrics
|
|
|
|
|
|
@pytest.mark.skipif(sys.platform == "darwin", reason="do not run on mac os")
|
|
def test_custom_metric():
|
|
from flaml import AutoML
|
|
import requests
|
|
|
|
X_train, y_train, X_val, y_val, X_test = get_toy_data_seqclassification()
|
|
automl = AutoML()
|
|
|
|
try:
|
|
import ray
|
|
|
|
if not ray.is_initialized():
|
|
ray.init()
|
|
except ImportError:
|
|
return
|
|
|
|
automl_settings = get_automl_settings()
|
|
automl_settings["metric"] = custom_metric
|
|
automl_settings["use_ray"] = {"local_dir": "data/output/"}
|
|
|
|
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
|
|
|
|
# testing calling custom metric in TransformersEstimator._compute_metrics_by_dataset_name
|
|
|
|
automl_settings["max_iter"] = 3
|
|
automl.fit(
|
|
X_train=X_train, y_train=y_train, X_val=X_val, y_val=y_val, **automl_settings
|
|
)
|
|
automl.score(X_val, y_val, **{"metric": custom_metric})
|
|
automl.pickle("automl.pkl")
|
|
|
|
del automl
|
|
|
|
if os.path.exists("test/data/output/"):
|
|
try:
|
|
shutil.rmtree("test/data/output/")
|
|
except PermissionError:
|
|
print("PermissionError when deleting test/data/output/")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_custom_metric()
|