diff --git a/colabs/boosting/Simple_LightGBM_Integration.ipynb b/colabs/boosting/Simple_LightGBM_Integration.ipynb index 95222f8f..e84f4e82 100644 --- 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)" + " )" ] }, {