Doctorado-ML/stree_datasets
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/Stree_datasets.git synced 2025-08-16 07:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix kernel name in Models

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2021-01-01 00:48:07 +01:00
parent e089f8c972
commit 2bbf34ff43
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
experimentation/Models.py
View File

@@ -56,7 +56,7 @@ class ModelStree(ModelBase):
"split_criteria": split_criteria, "split_criteria": split_criteria,
"max_features": max_features, "max_features": max_features,
} }
self._poly = { self._rbf = {
"random_state": [self._random_state], "random_state": [self._random_state],
"kernel": ["rbf"], "kernel": ["rbf"],
"C": C, "C": C,
@@ -65,7 +65,7 @@ class ModelStree(ModelBase):
"split_criteria": split_criteria, "split_criteria": split_criteria,
"max_features": max_features, "max_features": max_features,
} }
self._rbf = { self._poly = {
"random_state": [self._random_state], "random_state": [self._random_state],
"kernel": ["poly"], "kernel": ["poly"],
"degree": [3, 5], "degree": [3, 5],