github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-04-20 03:02:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

pickle the AutoML object (#37)

Browse Source 
* pickle the AutoML object

* get best model per estimator

* test deberta

* stateless API

* Add Gitter badge (#41)

* prevent divide by zero

* test roberta

* BlendSearchTuner

Co-authored-by: Chi Wang (MSR) <chiw@microsoft.com>
Co-authored-by: The Gitter Badger <badger@gitter.im>
This commit is contained in:
Chi Wang
2021-03-16 22:13:35 -07:00
committed by GitHub
parent ec37ae8f8f
commit 4a8110c87b
19 changed files with 685 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
notebook/flaml_automl.ipynb
View File

@@ -385,10 +385,10 @@
},
"outputs": [],
"source": [
"''' pickle and save the best model '''\n",
"''' pickle and save the automl object '''\n",
"import pickle\n",
"with open('best_model.pkl', 'wb') as f:\n",
" pickle.dump(automl.model, f, pickle.HIGHEST_PROTOCOL)"
"with open('automl.pkl', 'wb') as f:\n",
" pickle.dump(automl, f, pickle.HIGHEST_PROTOCOL)"
]
},
{

6
notebook/flaml_azureml.ipynb
View File

@@ -302,10 +302,10 @@
},
"outputs": [],
"source": [
"''' pickle and save the best model '''\n",
"''' pickle and save the automl object '''\n",
"import pickle\n",
"with open('best_model.pkl', 'wb') as f:\n",
" pickle.dump(automl.model, f, pickle.HIGHEST_PROTOCOL)"
"with open('automl.pkl', 'wb') as f:\n",
" pickle.dump(automl, f, pickle.HIGHEST_PROTOCOL)"
]
},
{

6
notebook/flaml_lightgbm.ipynb
View File

@@ -273,10 +273,10 @@
},
"outputs": [],
"source": [
"''' pickle and save the best model '''\n",
"''' pickle and save the automl object '''\n",
"import pickle\n",
"with open('best_model.pkl', 'wb') as f:\n",
" pickle.dump(automl.model, f, pickle.HIGHEST_PROTOCOL)"
"with open('automl.pkl', 'wb') as f:\n",
" pickle.dump(automl, f, pickle.HIGHEST_PROTOCOL)"
]
},
{

6
notebook/flaml_xgboost.ipynb
View File

@@ -321,10 +321,10 @@
},
"outputs": [],
"source": [
"''' pickle and save the best model '''\n",
"''' pickle and save the automl object '''\n",
"import pickle\n",
"with open('best_model.pkl', 'wb') as f:\n",
" pickle.dump(automl.model, f, pickle.HIGHEST_PROTOCOL)"
"with open('automl.pkl', 'wb') as f:\n",
" pickle.dump(automl, f, pickle.HIGHEST_PROTOCOL)"
]
},
{