From 6095be039ff6f45f82078120da0109aea82b66f5 Mon Sep 17 00:00:00 2001 From: Chi Wang Date: Tue, 15 Feb 2022 13:17:43 -0800 Subject: [PATCH] update regression metrics in notebooks (#454) --- notebook/automl_lightgbm.ipynb | 4 ++-- notebook/automl_xgboost.ipynb | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/notebook/automl_lightgbm.ipynb b/notebook/automl_lightgbm.ipynb index a661f1000..29ab0c9f3 100644 --- a/notebook/automl_lightgbm.ipynb +++ b/notebook/automl_lightgbm.ipynb @@ -122,7 +122,7 @@ "source": [ "settings = {\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", " \"task\": 'regression', # task type \n", " \"log_file_name\": 'houses_experiment.log', # flaml log file\n", @@ -977,7 +977,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.8.0" + "version": "3.9.7" } }, "nbformat": 4, diff --git a/notebook/automl_xgboost.ipynb b/notebook/automl_xgboost.ipynb index e5b41f167..7cabe86bc 100644 --- a/notebook/automl_xgboost.ipynb +++ b/notebook/automl_xgboost.ipynb @@ -121,7 +121,7 @@ "source": [ "settings = {\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", " \"task\": 'regression', # task type \n", " \"log_file_name\": 'houses_experiment.log', # flaml log file\n",