model trainign

experiments/german_credit.jl

 counterfactual_data, test_data = train_test_split(load_german_credit(nothing); test_size=TEST_SIZE)
 
 # 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),
@@ -10,7 +10,7 @@ builder = MLJFlux.@builder Flux.Chain(
 )
 
 # Number of individuals:
-n_ind = N_IND_SPECIFIED ? N_IND : 10
+n_ind = N_IND_SPECIFIED ? N_IND : 100
 
 run_experiment(
     counterfactual_data, test_data;
@@ -20,8 +20,10 @@ run_experiment(
     sampling_batch_size=10,
     sampling_steps=30,
     use_ensembling=true,
-    Λ=[0.1, 0.5, 0.5],
     opt=Flux.Optimise.Descent(0.05),
     n_individuals=n_ind,
-    use_variants=false
+    use_variants=true,
+    Λ=[0.1, 0.2, 0.2],
+    nsamples=100,
+    niter_eccco=100
 )
\ No newline at end of file

experiments/jobscripts/models/german_credit.sh

+#!/bin/bash
+
+#SBATCH --job-name="Train German Credit (ECCCo)"
+#SBATCH --time=01:00:00
+#SBATCH --ntasks=1
+#SBATCH --gpus-per-task=1
+#SBATCH --cpus-per-task=1
+#SBATCH --partition=compute
+#SBATCH --mem-per-cpu=4GB
+#SBATCH --account=innovation
+#SBATCH --mail-type=END     # Set mail type to 'END' to receive a mail when the job finishes. 
+
+srun julia --project=experiments experiments/run_experiments.jl -- data=german_credit output_path=results only_models > experiments/german_credit.log
\ No newline at end of file

experiments/jobscripts/models/mnist.sh

+#!/bin/bash
+
+#SBATCH --job-name="Train MNIST (ECCCo)"
+#SBATCH --time=01:00:00
+#SBATCH --ntasks=1
+#SBATCH --gpus-per-task=1
+#SBATCH --cpus-per-task=1
+#SBATCH --partition=compute
+#SBATCH --mem-per-cpu=4GB
+#SBATCH --account=innovation
+#SBATCH --mail-type=END     # Set mail type to 'END' to receive a mail when the job finishes. 
+
+srun julia --project=experiments experiments/run_experiments.jl -- data=mnist output_path=results only_models > experiments/mnist.log
\ No newline at end of file