Skip to content

Commit 058300d

Browse files
Merge pull request #2106 from recommenders-team/bug/log_loss_breaking_change
Fix breaking change in sklearn in log_loss
2 parents d849dc6 + a92b31a commit 058300d

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

examples/00_quick_start/lightgbm_tinycriteo.ipynb

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -717,7 +717,7 @@
717717
"source": [
718718
"test_preds = lgb_model.predict(test_x)\n",
719719
"auc = roc_auc_score(np.asarray(test_y.reshape(-1)), np.asarray(test_preds))\n",
720-
"logloss = log_loss(np.asarray(test_y.reshape(-1)), np.asarray(test_preds), eps=1e-12)\n",
720+
"logloss = log_loss(np.asarray(test_y.reshape(-1)), np.asarray(test_preds))\n",
721721
"res_basic = {\"auc\": auc, \"logloss\": logloss}\n",
722722
"print(res_basic)\n"
723723
]
@@ -904,7 +904,7 @@
904904
],
905905
"source": [
906906
"auc = roc_auc_score(np.asarray(test_y.reshape(-1)), np.asarray(test_preds))\n",
907-
"logloss = log_loss(np.asarray(test_y.reshape(-1)), np.asarray(test_preds), eps=1e-12)\n",
907+
"logloss = log_loss(np.asarray(test_y.reshape(-1)), np.asarray(test_preds))\n",
908908
"res_optim = {\"auc\": auc, \"logloss\": logloss}\n",
909909
"\n",
910910
"print(res_optim)"
@@ -959,7 +959,7 @@
959959
],
960960
"source": [
961961
"auc = roc_auc_score(np.asarray(test_y.reshape(-1)), np.asarray(test_preds))\n",
962-
"logloss = log_loss(np.asarray(test_y.reshape(-1)), np.asarray(test_preds), eps=1e-12)\n",
962+
"logloss = log_loss(np.asarray(test_y.reshape(-1)), np.asarray(test_preds))\n",
963963
"\n",
964964
"print({\"auc\": auc, \"logloss\": logloss})"
965965
]

0 commit comments

Comments
 (0)