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

Add select KBest features #17 (#35)

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2021-04-26 01:48:50 +02:00
committed by GitHub
parent a4aac9d310
commit 02de394c96
3 changed files with 65 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
stree/tests/Stree_test.py
View File

@@ -315,7 +315,7 @@ class Stree_test(unittest.TestCase):
X, y = load_dataset(self._random_state)
clf = Stree(random_state=self._random_state, max_features=2)
clf.fit(X, y)
self.assertAlmostEqual(0.9246666666666666, clf.score(X, y))
self.assertAlmostEqual(0.9453333333333334, clf.score(X, y))
def test_bogus_splitter_parameter(self):
clf = Stree(splitter="duck")