From 045e2fd446195f872e0bfe4eb275364cd6eaeacd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= Date: Mon, 19 Apr 2021 12:28:54 +0200 Subject: [PATCH] Fix random_sate issue in non linear kernels --- stree/Strees.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/stree/Strees.py b/stree/Strees.py index 84a39e9..14d0406 100644 --- a/stree/Strees.py +++ b/stree/Strees.py @@ -734,7 +734,7 @@ class Stree(BaseEstimator, ClassifierMixin): return node def _build_clf(self): - """Build the correct classifier for the node""" + """Build the right classifier for the node""" return ( LinearSVC( max_iter=self.max_iter, @@ -750,6 +750,7 @@ class Stree(BaseEstimator, ClassifierMixin): C=self.C, gamma=self.gamma, degree=self.degree, + random_state=self.random_state, ) )