Merge branch 'tsvd' of https://github.com/norm4nn/scholar into tsvd

elixir-nx · Oct 1, 2024 · 90856f5 · 90856f5
2 parents 6b57e1a + aa30ade
commit 90856f5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lib/scholar/decomposition/pca.ex b/lib/scholar/decomposition/pca.ex
@@ -471,6 +471,7 @@ defmodule Scholar.Decomposition.PCA do
     end
   end
 
+  @doc false
   defn flip_svd(u, v) do
     max_abs_cols_idx = u |> Nx.abs() |> Nx.argmax(axis: 0, keep_axis: true)
     signs = u |> Nx.take_along_axis(max_abs_cols_idx, axis: 0) |> Nx.sign() |> Nx.squeeze()