Combinatorial explosion (#19)

* Remove itertools combinations from subspaces

* Generates 5 random subspaces at most
This commit is contained in:
Ricardo Montañana Gómez
2021-01-10 13:32:22 +01:00
committed by GitHub
parent 475ad7e752
commit 36816074ff
3 changed files with 38 additions and 17 deletions

View File

@@ -313,7 +313,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.944, clf.score(X, y))
self.assertAlmostEqual(0.9246666666666666, clf.score(X, y))
def test_bogus_splitter_parameter(self):
clf = Stree(splitter="duck")