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

Fix entroy and information_gain functions

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2020-06-16 13:56:02 +02:00
parent a20e45e8e7
commit 3e52a4746c
3 changed files with 156 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
stree/tests/Stree_test.py
View File

@@ -239,6 +239,9 @@ class Stree_test(unittest.TestCase):
"min_distance linear": 0.9533333333333334,
"min_distance rbf": 0.836,
"min_distance poly": 0.9473333333333334,
"max_distance linear": 0.9533333333333334,
"max_distance rbf": 0.836,
"max_distance poly": 0.9473333333333334,
},
"Iris": {
"max_samples linear": 0.98,
@@ -247,11 +250,14 @@ class Stree_test(unittest.TestCase):
"min_distance linear": 0.98,
"min_distance rbf": 1.0,
"min_distance poly": 1.0,
"max_distance linear": 0.98,
"max_distance rbf": 1.0,
"max_distance poly": 1.0,
},
}
for name, dataset in datasets.items():
px, py = dataset
for criteria in ["max_samples", "min_distance"]:
for criteria in ["max_samples", "min_distance", "max_distance"]:
for kernel in self._kernels:
clf = Stree(
C=1e4,