yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-12-06 03:47:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

update regression metrics in notebooks (#454)

Browse Source
This commit is contained in:
Chi Wang 2022-02-15 13:17:43 -08:00 committed by GitHub
parent 05f9065ade
commit 6095be039f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
notebook/automl_lightgbm.ipynb
View File

@ -122,7 +122,7 @@
"source": [ "source": [
"settings = {\n", "settings = {\n",
" \"time_budget\": 240, # total running time in seconds\n", " \"time_budget\": 240, # total running time in seconds\n",
" \"metric\": 'r2', # primary metrics for regression can be chosen from: ['mae','mse','r2']\n", " \"metric\": 'r2', # primary metrics for regression can be chosen from: ['mae','mse','r2','rmse','mape']\n",
" \"estimator_list\": ['lgbm'], # list of ML learners; we tune lightgbm in this example\n", " \"estimator_list\": ['lgbm'], # list of ML learners; we tune lightgbm in this example\n",
" \"task\": 'regression', # task type \n", " \"task\": 'regression', # task type \n",
" \"log_file_name\": 'houses_experiment.log', # flaml log file\n", " \"log_file_name\": 'houses_experiment.log', # flaml log file\n",
@ -977,7 +977,7 @@
"name": "python", "name": "python",
"nbconvert_exporter": "python", "nbconvert_exporter": "python",
"pygments_lexer": "ipython3", "pygments_lexer": "ipython3",
"version": "3.8.0" "version": "3.9.7"
} }
}, },
"nbformat": 4, "nbformat": 4,

2
notebook/automl_xgboost.ipynb
View File

@ -121,7 +121,7 @@
"source": [ "source": [
"settings = {\n", "settings = {\n",
" \"time_budget\": 120, # total running time in seconds\n", " \"time_budget\": 120, # total running time in seconds\n",
" \"metric\": 'r2', # primary metrics for regression can be chosen from: ['mae','mse','r2']\n", " \"metric\": 'r2', # primary metrics for regression can be chosen from: ['mae','mse','r2','rmse','mape']\n",
" \"estimator_list\": ['xgboost'], # list of ML learners; we tune xgboost in this example\n", " \"estimator_list\": ['xgboost'], # list of ML learners; we tune xgboost in this example\n",
" \"task\": 'regression', # task type \n", " \"task\": 'regression', # task type \n",
" \"log_file_name\": 'houses_experiment.log', # flaml log file\n", " \"log_file_name\": 'houses_experiment.log', # flaml log file\n",