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, ) )