Skip to content

Commit 7241c8f

Browse files
committed
update benchmark interpret-core version
1 parent a1a54c4 commit 7241c8f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/benchmarks/ebm-benchmark.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
"force_recreate = False\n",
1818
"exist_ok = True\n",
1919
"TIMEOUT_SEC = 60 * 60 * 24 * 180 # 180 days\n",
20-
"wheel_filepaths = ['interpret_core-0.6.15-py3-none-any.whl', 'powerlift-0.1.12-py3-none-any.whl']\n",
20+
"wheel_filepaths = ['interpret_core-0.7.1-py3-none-any.whl', 'powerlift-0.1.12-py3-none-any.whl']\n",
2121
"\n",
2222
"import datetime\n",
2323
"experiment_name = datetime.datetime.now().strftime('%Y_%m_%d_%H%M__') + 'myexperiment'\n",
@@ -834,7 +834,7 @@
834834
" print(f\"FIT: {global_counter}, {trial.task.origin}, {trial.task.name}, {trial.method}, {trial.meta}, classes:{trial.task.n_classes}, features:{fit_params['X'].shape[1]}, train_samples:{fit_params['X'].shape[0]}, orig_samples:{trial.task.n_samples}\")\n",
835835
"\n",
836836
" if isinstance(est, (ExplainableBoostingClassifier, ExplainableBoostingRegressor)):\n",
837-
" n_bytes = est.estimate_mem(fit_params[\"X\"])\n",
837+
" n_bytes = est.estimate_mem(fit_params[\"X\"], fit_params[\"y\"])\n",
838838
" print(f\"EBM Memory Required: {n_bytes}\")\n",
839839
" trial.log(\"mem\", n_bytes)\n",
840840
" \n",

0 commit comments

Comments
 (0)