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

Fix failing tests

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2020-05-17 18:55:33 +02:00
parent 91af875062
commit 3cd004350d
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/Stree_test.py
View File

@@ -133,7 +133,7 @@ class Stree_test(unittest.TestCase):
def test_score(self): def test_score(self):
X, y = self._get_Xy() X, y = self._get_Xy()
accuracy_score = self._clf.score(X, y, print_out=False) accuracy_score = self._clf.score(X, y)
yp = self._clf.predict(X) yp = self._clf.predict(X)
right = (yp == y).astype(int) right = (yp == y).astype(int)
accuracy_computed = sum(right) / len(y) accuracy_computed = sum(right) / len(y)
@@ -178,8 +178,8 @@ class Stree_test(unittest.TestCase):
model_predictions.predict(X).tolist(), model_predictions.predict(X).tolist(),
model_hyperplane.predict(X).tolist() model_hyperplane.predict(X).tolist()
) )
a = model_predictions.score(X, y, print_out=False), a = model_predictions.score(X, y),
b = model_hyperplane.score(X, y, print_out=False) b = model_hyperplane.score(X, y)
self.assertEqual(a, b) self.assertEqual(a, b)
self.assertGreater(b, .95) self.assertGreater(b, .95)