From 44c6b3fcc890142b900cfffc1553efdec8efaf77 Mon Sep 17 00:00:00 2001
From: skfaysal <faysal.dsl.ai@gmail.com>
Date: Wed, 13 Dec 2023 23:28:31 +0600
Subject: [PATCH 1/2] distance parameter added on KMedoids
---
sklearn_extra/cluster/_k_medoids.py | 10 +++++++++-
1 file changed, 9 insertions(+), 1 deletion(-)
diff --git a/sklearn_extra/cluster/_k_medoids.py b/sklearn_extra/cluster/_k_medoids.py
index bb5165ba..9b5f93ca 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
From e1e01648781ca57557eafff8fd37c51e47bffb53 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
<66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Wed, 13 Dec 2023 17:41:55 +0000
Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
---
sklearn_extra/cluster/_k_medoids.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/sklearn_extra/cluster/_k_medoids.py b/sklearn_extra/cluster/_k_medoids.py
index 9b5f93ca..a5e4f223 100644
--- a/sklearn_extra/cluster/_k_medoids.py
+++ b/sklearn_extra/cluster/_k_medoids.py
@@ -315,11 +315,11 @@ def fit(self, X, y=None):
# Extra Addition
distances = self.transform(X)
self.distances_ = distances
-
+
# Modification
- self.inertia_ = _compute_inertia(distances) ## Prebiously this line was _compute_inertia(self.transform(X))
-
-
+ self.inertia_ = _compute_inertia(
+ distances
+ ) ## Prebiously this line was _compute_inertia(self.transform(X))
# Return self to enable method chaining
return self