github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-02-16 08:45:06 -05:00
Code Issues Packages Projects Releases Wiki Activity

Quick-fix (#539)

Browse Source 
* fix doc string; enable label transform in automl.score
This commit is contained in:
Qiaochu Song
2022-05-19 11:43:34 -04:00
committed by GitHub
parent 7126b69ce0
commit 2851134052
3 changed files with 60 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
test/automl/test_score.py
View File

@@ -212,6 +212,64 @@ class TestScore:
except NotImplementedError:
pass
def test_class(self):
# to test classification task with labels need encoding
X = pd.DataFrame(
{
"f1": [1, -2, 3, -4, 5, -6, -7, 8, -9, -10, -11, -12, -13, -14],
"f2": [
3.0,
16.0,
10.0,
12.0,
3.0,
14.0,
11.0,
12.0,
5.0,
14.0,
20.0,
16.0,
15.0,
11.0,
],
}
)
y = pd.Series(
[
"a",
"b",
"c",
"d",
"a",
"b",
"c",
"d",
"a",
"b",
"c",
"d",
"a",
"b",
]
)
automl = AutoML()
automl_settings = {
"time_budget": 6,
"task": "classification",
"n_jobs": 1,
"estimator_list": ["xgboost"],
"metric": "accuracy",
"log_training_metric": True,
}
automl.fit(X, y, **automl_settings)
assert automl._label_transformer is not None
assert automl.score(X, y) > 0
automl.pickle("automl.pkl")
if __name__ == "__main__":
test = TestScore()