Pin LGB 4.0 (#537)

* import * fix for lgb 4.0
wandb · Jun 18, 2024 · 0c7ed7d · 0c7ed7d
1 parent 21dff55
commit 0c7ed7d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/colabs/boosting/Simple_LightGBM_Integration.ipynb b/colabs/boosting/Simple_LightGBM_Integration.ipynb
@@ -65,7 +65,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install -Uq 'lightgbm>=3.3.1'"
+    "!pip install -Uq 'lightgbm>=4.0.0'"
    ]
   },
   {
@@ -202,7 +202,7 @@
     "    'verbosity': 0\n",
     "}\n",
     "\n",
-    "wandb.init(project='my-lightgbm-project', config=params);"
+    "wandb.init(project='my-lightgbm-project', config=params)"
    ]
   },
   {
@@ -234,8 +234,9 @@
     "                num_boost_round=30,\n",
     "                valid_sets=lgb_eval,\n",
     "                valid_names=('validation'),\n",
+    "                early_stopping_rounds=5\n",
     "                callbacks=[wandb_callback()],\n",
-    "                early_stopping_rounds=5)"
+    "                )"
    ]
   },
   {