Update lib/scholar/linear/bayesian_ridge_regression.ex

Co-authored-by: Krsto Proroković <krstopro@yahoo.com>
elixir-nx · Apr 5, 2024 · 3694a0d · 3694a0d
1 parent 287eb6a
commit 3694a0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/scholar/linear/bayesian_ridge_regression.ex b/lib/scholar/linear/bayesian_ridge_regression.ex
@@ -186,7 +186,7 @@ defmodule Scholar.Linear.BayesianRidgeRegression do
 
     iterations = opts[:iterations]
 
-    xt_y = Nx.dot(Nx.transpose(x), y)
+xt_y = Nx.dot(x, [0], y, [0])
     {u, s, vh} = Nx.LinAlg.svd(x, full_matrices?: false)
     eigenvals = Nx.pow(s, 2)
     {n_samples, n_features} = Nx.shape(x)