From a92b31a644c91550e696a0a800073e8689de3ea7 Mon Sep 17 00:00:00 2001 From: miguelgfierro Date: Mon, 3 Jun 2024 16:58:06 +0200 Subject: [PATCH] breaking change in sklearn in log_loss :boom::boom: Signed-off-by: miguelgfierro --- examples/00_quick_start/lightgbm_tinycriteo.ipynb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/00_quick_start/lightgbm_tinycriteo.ipynb b/examples/00_quick_start/lightgbm_tinycriteo.ipynb index f7a786415..ffd827eac 100644 --- a/examples/00_quick_start/lightgbm_tinycriteo.ipynb +++ b/examples/00_quick_start/lightgbm_tinycriteo.ipynb @@ -717,7 +717,7 @@ "source": [ "test_preds = lgb_model.predict(test_x)\n", "auc = roc_auc_score(np.asarray(test_y.reshape(-1)), np.asarray(test_preds))\n", - "logloss = log_loss(np.asarray(test_y.reshape(-1)), np.asarray(test_preds), eps=1e-12)\n", + "logloss = log_loss(np.asarray(test_y.reshape(-1)), np.asarray(test_preds))\n", "res_basic = {\"auc\": auc, \"logloss\": logloss}\n", "print(res_basic)\n" ] @@ -904,7 +904,7 @@ ], "source": [ "auc = roc_auc_score(np.asarray(test_y.reshape(-1)), np.asarray(test_preds))\n", - "logloss = log_loss(np.asarray(test_y.reshape(-1)), np.asarray(test_preds), eps=1e-12)\n", + "logloss = log_loss(np.asarray(test_y.reshape(-1)), np.asarray(test_preds))\n", "res_optim = {\"auc\": auc, \"logloss\": logloss}\n", "\n", "print(res_optim)" @@ -959,7 +959,7 @@ ], "source": [ "auc = roc_auc_score(np.asarray(test_y.reshape(-1)), np.asarray(test_preds))\n", - "logloss = log_loss(np.asarray(test_y.reshape(-1)), np.asarray(test_preds), eps=1e-12)\n", + "logloss = log_loss(np.asarray(test_y.reshape(-1)), np.asarray(test_preds))\n", "\n", "print({\"auc\": auc, \"logloss\": logloss})" ]