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

remove unneed test and cosmetic

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2020-06-06 14:20:07 +02:00
parent b9f14aec05
commit cb10aea36e
2 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
stree/Strees.py
View File

@@ -192,10 +192,9 @@ class Stree(BaseEstimator, ClassifierMixin):
if self.use_predictions: if self.use_predictions:
res = np.expand_dims(node._clf.decision_function(data), 1) res = np.expand_dims(node._clf.decision_function(data), 1)
else: else:
"""doesn't work with multiclass as each sample has to do inner # doesn't work with multiclass as each sample has to do inner
product with its own coefficients computes positition of every # product with its own coefficients computes positition of every
sample is w.r.t. the hyperplane # sample is w.r.t. the hyperplane
"""
res = self._linear_function(data, node) res = self._linear_function(data, node)
return res return res

3
stree/tests/Strees_grapher_test.py
View File

@@ -95,9 +95,6 @@ class Stree_grapher_test(unittest.TestCase):
num_figures_after = plt.gcf().number num_figures_after = plt.gcf().number
self.assertEqual(7, num_figures_after - num_figures_before) self.assertEqual(7, num_figures_after - num_figures_before)
def test_filt_4_dims_dataset(self):
self._clf.fit(*get_dataset(self._random_state, n_features=4))
class Snode_graph_test(unittest.TestCase): class Snode_graph_test(unittest.TestCase):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):