Merge pull request #25 from CausalAIBook/fixed_flaml_update

Update python-nonlinear-ml-for-wage-prediction.ipynb
CausalAIBook · Sep 27, 2024 · 1c50e67 · 1c50e67
2 parents 98e421e + 2630d0e
commit 1c50e67
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/PM3/python-nonlinear-ml-for-wage-prediction.ipynb b/PM3/python-nonlinear-ml-for-wage-prediction.ipynb
@@ -1313,6 +1313,7 @@
    "source": [
     "automl = make_pipeline(base, AutoML(task='regression', time_budget=60, early_stop=True,\n",
     "                                    eval_method='cv', n_splits=3, metric='r2',\n",
+    "                                    estimator_list=['lgbm', 'xgboost', 'xgb_limitdepth', 'rf', 'extra_tree'],\n",
     "                                    verbose=3,\n",
     "                                    ensemble={'passthrough': False,  # whether stacker will use raw X's or predictions\n",
     "                                              'final_estimator': RLasso()}))"
@@ -1361,6 +1362,7 @@
    "source": [
     "automl = make_pipeline(base, AutoML(task='regression', time_budget=60, early_stop=True,\n",
     "                                    eval_method='cv', n_splits=3, metric='r2',\n",
+    "                                    estimator_list=['lgbm', 'xgboost', 'xgb_limitdepth', 'rf', 'extra_tree'],\n",
     "                                    verbose=3,\n",
     "                                    ensemble={'passthrough': True,  # whether stacker will use raw X's or predictions\n",
     "                                              'final_estimator': RLasso()}))"
@@ -1420,9 +1422,9 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.5"
+   "version": "3.12.4"
   }
  },
  "nbformat": 4,
- "nbformat_minor": 1
+ "nbformat_minor": 4
 }