Update benchmark hyperparams os STree

2025-08-15 23:46:02 +00:00 · 2021-10-31 12:41:30 +01:00
parent 8daecc4726
commit e5d49132ec
1 changed files with 2 additions and 2 deletions
--- a/notebooks/benchmark.ipynb
+++ b/notebooks/benchmark.ipynb
@@ -178,7 +178,7 @@
   "outputs": [],
   "source": [
    "# Stree\n",
-    "stree = Stree(random_state=random_state, C=.01, max_iter=1e3)"
+    "stree = Stree(random_state=random_state, C=.01, max_iter=1e3, kernel=\"liblinear\", multiclass_strategy=\"ovr\")"
   ]
  },
  {
@@ -368,4 +368,4 @@
 },
 "nbformat": 4,
 "nbformat_minor": 4
-}
+}