Reduce redundant code

elixir-nx · Oct 18, 2023 · 0e08f6a · 0e08f6a
1 parent 42a5b24
commit 0e08f6a
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/lib/scholar/cluster/affinity_propagation.ex b/lib/scholar/cluster/affinity_propagation.ex
@@ -315,10 +315,6 @@ defmodule Scholar.Cluster.AffinityPropagation do
       >
   """
   defn predict(%__MODULE__{cluster_centers: cluster_centers} = _model, x) do
-    {num_clusters, num_features} = Nx.shape(cluster_centers)
-    {num_samples, _} = Nx.shape(x)
-    broadcast_shape = {num_samples, num_clusters, num_features}
-
     dist = Scholar.Metrics.Distance.pairwise_euclidean(x, cluster_centers)
 
     Nx.select(Nx.is_nan(dist), Nx.Constants.infinity(Nx.type(dist)), dist)