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

Implement predict and score methods & tests

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2020-05-13 12:42:09 +02:00
parent c4de782a3f
commit 8f71eeb316
6 changed files with 70 additions and 221 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/Snode_test.py
View File

@@ -11,9 +11,9 @@ class Snode_test(unittest.TestCase):
def __init__(self, *args, **kwargs):
self._random_state = 1
self._model = Stree(random_state=self._random_state,
self._clf = Stree(random_state=self._random_state,
use_predictions=True)
self._model.fit(*self._get_Xy())
self._clf.fit(*self._get_Xy())
super(Snode_test, self).__init__(*args, **kwargs)
def _get_Xy(self):
@@ -42,4 +42,4 @@ class Snode_test(unittest.TestCase):
return
check_leave(node.get_down())
check_leave(node.get_up())
check_leave(self._model._tree)
check_leave(self._clf._tree)