diff --git a/sklearn_extra/cluster/_k_medoids.py b/sklearn_extra/cluster/_k_medoids.py index bb5165ba..a5e4f223 100644 --- a/sklearn_extra/cluster/_k_medoids.py +++ b/sklearn_extra/cluster/_k_medoids.py @@ -311,7 +311,15 @@ def fit(self, X, y=None): # the training data to clusters self.labels_ = np.argmin(D[medoid_idxs, :], axis=0) self.medoid_indices_ = medoid_idxs - self.inertia_ = _compute_inertia(self.transform(X)) + + # Extra Addition + distances = self.transform(X) + self.distances_ = distances + + # Modification + self.inertia_ = _compute_inertia( + distances + ) ## Prebiously this line was _compute_inertia(self.transform(X)) # Return self to enable method chaining return self