Remove redundant sorting (#220)

elixir-nx · Dec 14, 2023 · 6438349 · 6438349
1 parent acb0673
commit 6438349
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/scholar/naive_bayes/gaussian.ex b/lib/scholar/naive_bayes/gaussian.ex
@@ -273,7 +273,7 @@ defmodule Scholar.NaiveBayes.Gaussian do
     num_classes = opts[:num_classes]
     priors_flag = opts[:priors_flag]
 
-    classes = Nx.iota({num_classes}) |> Nx.sort()
+    classes = Nx.iota({num_classes})
 
     class_priors =
       case Nx.shape(class_priors) do