From e5d49132ec884ead078053080029777ace9cf8f5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= Date: Sun, 31 Oct 2021 12:41:30 +0100 Subject: [PATCH] Update benchmark hyperparams os STree --- notebooks/benchmark.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/notebooks/benchmark.ipynb b/notebooks/benchmark.ipynb index edc3091..8dbb2d4 100644 --- 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 -} \ No newline at end of file +}