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",