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

Add a method that return nodes and leaves (#27) (#30)

Browse Source 
Add a test
Fix #27
This commit is contained in:
Ricardo Montañana Gómez
2021-03-23 14:30:32 +01:00
committed by GitHub
parent 460c63a6d0
commit 6ba973dfe1
2 changed files with 36 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
stree/tests/Stree_test.py
View File

@@ -454,3 +454,22 @@ class Stree_test(unittest.TestCase):
clf = Stree(random_state=self._random_state)
clf.fit(X, y)
self.assertEqual(7, clf.depth_)
def test_nodes_leaves(self):
X, y = load_dataset(
random_state=self._random_state,
n_classes=3,
n_features=5,
n_samples=1500,
)
clf = Stree(random_state=self._random_state)
clf.fit(X, y)
nodes, leaves = clf.nodes_leaves()
self.assertEqual(12, nodes)
self.assertEquals(13, leaves)
X, y = load_wine(return_X_y=True)
clf = Stree(random_state=self._random_state)
clf.fit(X, y)
nodes, leaves = clf.nodes_leaves()
self.assertEqual(8, nodes)
self.assertEquals(9, leaves)