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

Fix automl settings in scikit-learn pipeline integration example (#602)

Browse Source 
* Added test directory and core file to gitignore.

Closes #601.

* Fixed pipeline fit parameters.

Closes #600.

* Reverted changes to gitignore.
This commit is contained in:
Zvi Baratz
2022-06-23 05:44:14 +03:00
committed by GitHub
parent c45741a67b
commit fefdb48ef3
2 changed files with 211 additions and 202 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
website/docs/Examples/Integrate - Scikit-learn Pipeline.md
View File

@@ -37,16 +37,17 @@ automl_pipeline
### Run AutoML in the pipeline
```python
settings = {
automl_settings = {
"time_budget": 60, # total running time in seconds
"metric": 'accuracy', # primary metrics can be chosen from: ['accuracy','roc_auc', 'roc_auc_ovr', 'roc_auc_ovo', 'f1','log_loss','mae','mse','r2']
"task": 'classification', # task type
"estimator_list":['xgboost','catboost','lgbm'],
"log_file_name": 'airlines_experiment.log', # flaml log file
"metric": "accuracy", # primary metrics can be chosen from: ['accuracy','roc_auc', 'roc_auc_ovr', 'roc_auc_ovo', 'f1','log_loss','mae','mse','r2']
"task": "classification", # task type
"estimator_list": ["xgboost", "catboost", "lgbm"],
"log_file_name": "airlines_experiment.log", # flaml log file
}
automl_pipeline.fit(X_train, y_train,
automl__time_budget=60,
automl__metric="accuracy")
pipeline_settings = {
f"automl__{key}": value for key, value in automl_settings.items()
}
automl_pipeline.fit(X_train, y_train, **pipeline_settings)
```
### Get the automl object from the pipeline