diff options
-rw-r--r-- | src/python/gudhi/clustering/tomato.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/python/gudhi/clustering/tomato.py b/src/python/gudhi/clustering/tomato.py index e3d814d1..000fdf3d 100644 --- a/src/python/gudhi/clustering/tomato.py +++ b/src/python/gudhi/clustering/tomato.py @@ -312,13 +312,13 @@ class Tomato: else: self.labels_ = self.leaf_labels_ self.__n_clusters = self.n_leaves_ + return self def fit_predict(self, X, y=None, weights=None): """ Equivalent to fit(), and returns the `labels_`. """ - self.fit(X, y, weights) - return self.labels_ + return self.fit(X, y, weights).labels_ # TODO: add argument k or threshold? Have a version where you can click and it shows the line and the corresponding k? def plot_diagram(self): |