uff

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,
 )
 

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:

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,
 )
 

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,
 )
 

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:

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: