Skip to content

Commit b293b59

Browse files
committed
fix
1 parent 09210b4 commit b293b59

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

test/fit/ols-ridge-lasso-elnet.jl

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,12 @@ n, p = 500, 100
5656
@test nnz(θ_fista) == 12 # sparse
5757
@test nnz(θ_ista) == 12
5858

59+
# scale_penalty_with_samples
60+
lr_scaled = LassoRegression/n; fit_intercept=false,
61+
scale_penalty_with_samples = true)
62+
θ_scaled = fit(lr_scaled, X, y)
63+
@test θ_scaled θ_fista
64+
5965
# with intercept
6066
lr1 = LassoRegression(λ, penalize_intercept=true,
6167
scale_penalty_with_samples = false)
@@ -92,12 +98,6 @@ n, p = 500, 100
9298
@test isapprox(J(θ1_sk), 178.5, rtol=1e-3)
9399
@test nnz(θ1_sk) == 6
94100
end
95-
96-
# scale_penalty_with_samples
97-
lr_scaled = LassoRegression/n; fit_intercept=false,
98-
scale_penalty_with_samples = true)
99-
θ_scaled = fit(lr_scaled, X, y)
100-
@test θ_scaled θ_fista
101101
end
102102

103103
@testset "elnet" begin

0 commit comments

Comments
 (0)