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 +}