github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-02-04 10:25:41 -05:00
Code Issues Packages Projects Releases Wiki Activity

Multiclass (#99)

Browse Source 
* utility functions

* stepsize lower bound
This commit is contained in:
Chi Wang
2021-06-04 10:31:33 -07:00
committed by GitHub
parent 0d3a0bfab6
commit f7cf2ea45a
11 changed files with 80 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/test_automl.py
View File

@@ -260,6 +260,14 @@ class TestAutoML(unittest.TestCase):
X_train=X_train, y_train=y_train, metric='micro_f1', **automl_settings)
automl_experiment_macro.fit(
X_train=X_train, y_train=y_train, metric='macro_f1', **automl_settings)
estimator = automl_experiment_macro.model
y_pred = estimator.predict(X_train)
y_pred_proba = estimator.predict_proba(X_train)
from flaml.ml import norm_confusion_matrix, multi_class_curves
print(norm_confusion_matrix(y_train, y_pred))
from sklearn.metrics import roc_curve, precision_recall_curve
print(multi_class_curves(y_train, y_pred_proba, roc_curve))
print(multi_class_curves(y_train, y_pred_proba, precision_recall_curve))
def test_regression(self):
automl_experiment = AutoML()