fixed bug in tabular

experiments/california_housing.jl

@@ -12,16 +12,12 @@ tuning_params = DEFAULT_GENERATOR_TUNING
 params = (
     n_hidden=32,
     activation=Flux.relu,
-    builder=MLJFlux.@builder Flux.Chain(
-        Dense(n_in, n_hidden, activation),
-        Dense(n_hidden, n_hidden, activation),
-        Dense(n_hidden, n_out),
-    ),
-    α = [1.0, 1.0, 1e-1],
-    sampling_batch_size = 10,
-    sampling_steps = 30,
-    use_ensembling = true,
-    opt = Flux.Optimise.Descent(0.05)
+    builder=default_builder(n_hidden=32, n_layers=3, activation=Flux.relu),
+    α=[1.0, 1.0, 1e-1],
+    sampling_batch_size=10,
+    sampling_steps=30,
+    use_ensembling=true,
+    opt=Flux.Optimise.Descent(0.05)
 )
 
 if !GRID_SEARCH

experiments/credit_default.jl

@@ -10,13 +10,9 @@ tuning_params = DEFAULT_GENERATOR_TUNING
 
 # Parameter choices:
 params = (
-    n_hidden = 32,
-    activation = Flux.relu,
-    builder = MLJFlux.@builder Flux.Chain(
-        Dense(n_in, n_hidden, activation),
-        Dense(n_hidden, n_hidden, activation),
-        Dense(n_hidden, n_out),
-    ),
+    n_hidden=32,
+    activation=Flux.relu,
+    builder=default_builder(n_hidden=32, n_layers=3, activation=Flux.relu),
     α=[1.0, 1.0, 1e-1],
     sampling_batch_size=10,
     sampling_steps=30,

experiments/german_credit.jl

@@ -12,16 +12,12 @@ tuning_params = DEFAULT_GENERATOR_TUNING
 params = (
     n_hidden=32,
     activation=Flux.relu,
-    builder=MLJFlux.@builder Flux.Chain(
-        Dense(n_in, n_hidden, activation),
-        Dense(n_hidden, n_hidden, activation),
-        Dense(n_hidden, n_out),
-    ),
-    α = [1.0, 1.0, 1e-1],
-    sampling_batch_size = 10,
-    sampling_steps = 30,
-    use_ensembling = true,
-    opt = Flux.Optimise.Descent(0.05)
+    builder=default_builder(n_hidden=32, n_layers=3, activation=Flux.relu),
+    α=[1.0, 1.0, 1e-1],
+    sampling_batch_size=10,
+    sampling_steps=30,
+    use_ensembling=true,
+    opt=Flux.Optimise.Descent(0.05)
 )
 
 if !GRID_SEARCH

experiments/gmsc.jl

@@ -13,11 +13,7 @@ tuning_params = DEFAULT_GENERATOR_TUNING
 params = (
     n_hidden=32,
     activation=Flux.relu,
-    builder=MLJFlux.@builder Flux.Chain(
-        Dense(n_in, n_hidden, activation),
-        Dense(n_hidden, n_hidden, activation),
-        Dense(n_hidden, n_out),
-    ),
+    builder=default_builder(n_hidden=32, n_layers=3, activation=Flux.relu),
     α = [1.0, 1.0, 1e-1],
     sampling_batch_size = 10,
     sampling_steps = 30,