Fix random_sate issue in non linear kernels

This commit is contained in:
2021-04-19 12:28:54 +02:00
parent 2d6921f9a5
commit 045e2fd446

View File

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