diff --git a/src/TR.jl b/src/TR.jl index 425fb06..0d62df6 100644 --- a/src/TR.jl +++ b/src/TR.jl @@ -532,7 +532,7 @@ K-fold CV for the Ridge regression problem, using the 'SVD-trick' for calculatin function TRSegCVUpdateFair(X, y, lambdas, cv, bOld) n, p = size(X); -rmsecvman = zeros(length(lambdasu)); +rmsecvman = zeros(length(lambdas)); nfolds = length(unique(cv)); for i = 1:nfolds @@ -547,8 +547,8 @@ for i = 1:nfolds U, s, V = svd(Xs, full=false); - denom = broadcast(.+, broadcast(./, lambdasu, s'), s')'; - denom2 = broadcast(.+, ones(n-sum(inds)), broadcast(./, lambdasu', s.^2)); + denom = broadcast(.+, broadcast(./, lambdas, s'), s')'; + denom2 = broadcast(.+, ones(n-sum(inds)), broadcast(./, lambdas', s.^2)); # Calculating regression coefficients bcoeffs = V * broadcast(./, (U' * ys), denom) .+ bOld .- V * broadcast(./, V' * bOld, denom2);