From e0f2f92138aeeadfc1e52e9354efdab4a9980b96 Mon Sep 17 00:00:00 2001
From: Joakim Skogholt <onlinereg@skogholt.xyz>
Date: Sat, 13 May 2023 20:20:17 +0200
Subject: [PATCH] Fixed embarrasing bug in regularizationMatrix

---
 src/TR.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/TR.jl b/src/TR.jl
index 5b906b1..77fe209 100644
--- a/src/TR.jl
+++ b/src/TR.jl
@@ -62,7 +62,7 @@ elseif regType == "legendre"
     P, _    = plegendre(regParam1-1, p);   
     regMat[end-regParam1+1:end,:] = sqrt(regParam2) * P;
 elseif regType == "L2"
-    regMat = I(size(X,2));
+    regMat = I(p);
 elseif regType == "std"
     regMat = regParam2;
 end