Doctorado-ML/stree
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/STree.git synced 2025-08-16 16:06:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix random_sate issue in non linear kernels

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2021-04-19 12:28:54 +02:00
parent 2d6921f9a5
commit 045e2fd446
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
stree/Strees.py
View File

@@ -734,7 +734,7 @@ class Stree(BaseEstimator, ClassifierMixin):
return node return node
def _build_clf(self): def _build_clf(self):
"""Build the correct classifier for the node""" """Build the right classifier for the node"""
return ( return (
LinearSVC( LinearSVC(
max_iter=self.max_iter, max_iter=self.max_iter,
@@ -750,6 +750,7 @@ class Stree(BaseEstimator, ClassifierMixin):
C=self.C, C=self.C,
gamma=self.gamma, gamma=self.gamma,
degree=self.degree, degree=self.degree,
random_state=self.random_state,
) )
) )