Skip to content

Commit f97c846

Browse files
author
ZebinYang
committed
skip grid search cv in build leaf if param_dict is empty; version 0.2.4
1 parent f36ea22 commit f97c846

File tree

1 file changed

+0
-18
lines changed

1 file changed

+0
-18
lines changed

examples/demo.ipynb

-18
Original file line numberDiff line numberDiff line change
@@ -218,24 +218,6 @@
218218
"pred_test = clf.predict_proba(test_x)[:, 1]\n",
219219
"roc_auc_score(train_y, pred_train.ravel()), roc_auc_score(test_y, pred_test.ravel())"
220220
]
221-
},
222-
{
223-
"cell_type": "code",
224-
"execution_count": null,
225-
"metadata": {
226-
"ExecuteTime": {
227-
"start_time": "2021-12-27T12:27:37.691Z"
228-
}
229-
},
230-
"outputs": [],
231-
"source": [
232-
"clf = GLMTreeClassifier(max_depth=1, min_samples_leaf=50, reg_lambda=[1e4],\n",
233-
" n_split_grid=20, n_screen_grid=5, n_feature_search=10)\n",
234-
"clf.fit(train_x, train_y)\n",
235-
"pred_train = clf.predict_proba(train_x)[:, 1]\n",
236-
"pred_test = clf.predict_proba(test_x)[:, 1]\n",
237-
"roc_auc_score(train_y, pred_train.ravel()), roc_auc_score(test_y, pred_test.ravel())"
238-
]
239221
}
240222
],
241223
"metadata": {

0 commit comments

Comments
 (0)