diff --git a/experiments/california_housing.jl b/experiments/california_housing.jl index 8c811093f53ada9caa6c0810a7ad99d2ad3e655e..ffe8610aa2f0eb0110eb7ed33863632bdb7e3838 100644 --- a/experiments/california_housing.jl +++ b/experiments/california_housing.jl @@ -30,6 +30,8 @@ params = ( sampling_steps=30, use_ensembling=true, opt=Flux.Optimise.Descent(0.05), + Λ=[0.1, 0.1, 0.1], + n_individuals=25, dim_reduction=true, ) diff --git a/experiments/circles.jl b/experiments/circles.jl index 80c772718a5de01327169ba6a8bb2d48d64eb1ba..59facfe398d776e95fc241d8540018f68507f358 100644 --- a/experiments/circles.jl +++ b/experiments/circles.jl @@ -19,6 +19,8 @@ params = ( epochs=100, α=[1.0, 1.0, 1e-2], sampling_steps=30, + opt=Flux.Optimise.Descent(0.05), + Λ=[0.1, 0.1, 0.05], ) # Best grid search params: diff --git a/experiments/german_credit.jl b/experiments/german_credit.jl index e213740d0d9df6fd2032e5164da176c052566f78..4d5c170c3b0e7344c6886771ed238048f0d4fb42 100644 --- a/experiments/german_credit.jl +++ b/experiments/german_credit.jl @@ -30,6 +30,8 @@ params = ( sampling_steps=30, use_ensembling=true, opt=Flux.Optimise.Descent(0.05), + Λ=[0.2, 0.2, 0.2], + n_individuals=25, dim_reduction=true, ) diff --git a/experiments/gmsc.jl b/experiments/gmsc.jl index f2bbfe68420cd259a0357a9f3d0a6382cd73169c..e21b72eef4089eadf6deaf9d94c1016a4cd020d1 100644 --- a/experiments/gmsc.jl +++ b/experiments/gmsc.jl @@ -31,6 +31,8 @@ params = ( sampling_steps = 30, use_ensembling = true, opt=Flux.Optimise.Descent(0.05), + Λ = [0.1, 0.1, 0.1], + n_individuals=25, dim_reduction=true, ) diff --git a/experiments/linearly_separable.jl b/experiments/linearly_separable.jl index 60a5c29291f2cfe4d9b31b9c4dc877e20db7426a..5fed9e4c19f7fadd1500e4be6effd0d4595eb6ea 100644 --- a/experiments/linearly_separable.jl +++ b/experiments/linearly_separable.jl @@ -20,6 +20,8 @@ params = ( n_layers=3, activation=Flux.swish, epochs=100, + Λ=[0.1, 0.1, 0.05], + opt=Flux.Optimise.Descent(0.01), ) # Best grid search params: diff --git a/experiments/moons.jl b/experiments/moons.jl index 5c91e2100cb7950845666031b66ed63584a7f1da..f92f0648ed7107e6cdc0bb84e673360a848f6947 100644 --- a/experiments/moons.jl +++ b/experiments/moons.jl @@ -19,6 +19,8 @@ params = ( epochs=500, sampling_batch_size=10, sampling_steps=30, + Λ=[0.1, 0.1, 0.5], + opt=Flux.Optimise.Descent(0.01), ) # Best grid search params: