From cb10aea36e480f82cbd6e8a9ffcf992204c44c5d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= Date: Sat, 6 Jun 2020 14:20:07 +0200 Subject: [PATCH] remove unneed test and cosmetic --- stree/Strees.py | 7 +++---- stree/tests/Strees_grapher_test.py | 3 --- 2 files changed, 3 insertions(+), 7 deletions(-) diff --git a/stree/Strees.py b/stree/Strees.py index ecdc1df..0e8b4cb 100644 --- a/stree/Strees.py +++ b/stree/Strees.py @@ -192,10 +192,9 @@ class Stree(BaseEstimator, ClassifierMixin): if self.use_predictions: res = np.expand_dims(node._clf.decision_function(data), 1) else: - """doesn't work with multiclass as each sample has to do inner - product with its own coefficients computes positition of every - sample is w.r.t. the hyperplane - """ + # doesn't work with multiclass as each sample has to do inner + # product with its own coefficients computes positition of every + # sample is w.r.t. the hyperplane res = self._linear_function(data, node) return res diff --git a/stree/tests/Strees_grapher_test.py b/stree/tests/Strees_grapher_test.py index cd47593..e702dcc 100644 --- a/stree/tests/Strees_grapher_test.py +++ b/stree/tests/Strees_grapher_test.py @@ -95,9 +95,6 @@ class Stree_grapher_test(unittest.TestCase): num_figures_after = plt.gcf().number 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): def __init__(self, *args, **kwargs):