Doctorado-ML/stree
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/STree.git synced 2025-08-15 15:36:00 +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):
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)
right = (yp == y).astype(int)
accuracy_computed = sum(right) / len(y)
@@ -178,8 +178,8 @@ class Stree_test(unittest.TestCase):
model_predictions.predict(X).tolist(),
model_hyperplane.predict(X).tolist()
)
a = model_predictions.score(X, y, print_out=False),
b = model_hyperplane.score(X, y, print_out=False)
a = model_predictions.score(X, y),
b = model_hyperplane.score(X, y)
self.assertEqual(a, b)
self.assertGreater(b, .95)