notebook for milestone3

CS634Project_Milestone3_AyeThuzar.ipynb

@@ -9486,11 +9486,10 @@
       "cell_type": "code",
       "source": [
         "# Save LGBM baseline model\n",
-        "pickle.dump(LGBMR_HF, open('lgbm_base.pkl', 'wb'))\n",
+        "pickle.dump(reg_lgbm_baseline, open('lgbm_base.pkl', 'wb'))\n",
         "\n",
         "# Save LightGBM model optimized via Optuna\n",
-        "pickle.dump(LGBMR_best_HF, open('lgbm_optimized.pkl', 'wb'))\n",
-        ""
+        "pickle.dump(lgbmreg, open('lgbm_optimized.pkl', 'wb'))\n"
       ],
       "metadata": {
         "id": "UL0nntirX9xy"