gsmc model

experiments/gmsc.jl

 counterfactual_data, test_data = train_test_split(load_gmsc(nothing); test_size=TEST_SIZE)
+nobs = size(counterfactual_data.X, 2)
 
 # Default builder:
-n_hidden = 128
-activation = Flux.swish
+n_hidden = 32
+activation = Flux.relu
 builder = MLJFlux.@builder Flux.Chain(
     Dense(n_in, n_hidden, activation),
     Dense(n_hidden, n_hidden, activation),
@@ -24,4 +25,5 @@ run_experiment(
     opt = Flux.Optimise.Descent(0.05),
     n_individuals = n_ind,
     use_variants = false, 
+    min_batch_size = 250,
 )
\ No newline at end of file