diff --git a/src/TR.jl b/src/TR.jl index 36fbe64..2a0ac86 100644 --- a/src/TR.jl +++ b/src/TR.jl @@ -484,8 +484,8 @@ for j = 1:length(lambdas) Xs = Xdata .- mX; ys = ydata .- my; - betas = [Xs; sqrt(lambdas[j]) * I(p)] \ [ys; zeros(p,1)]; - rmsecvman[j] += sum((y[vec(inds)] - ((X[vec(inds),:] .- mX) * betas .+ my)).^2); + betas = [Xs; sqrt(lambdas[j]) * I(p)] \ [ys; zeros(p,1)]; + rmsecv[j] += sum((y[vec(inds)] - ((X[vec(inds),:] .- mX) * betas .+ my)).^2); end end