added cali housing

experiments/california_housing.jl

+counterfactual_data, test_data = train_test_split(load_california_housing(nothing); test_size=TEST_SIZE)
+nobs = size(counterfactual_data.X, 2)
+
+# Default builder:
+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),
+)
+
+# Number of individuals:
+n_ind = N_IND_SPECIFIED ? N_IND : 100
+
+run_experiment(
+    counterfactual_data, test_data;
+    dataname="California Housing",
+    epochs=100,
+    builder=builder,
+    α=[1.0, 1.0, 1e-1],
+    sampling_batch_size=10,
+    sampling_steps=30,
+    use_ensembling=true,
+    opt=Flux.Optimise.Descent(0.05),
+    n_individuals=n_ind,
+    min_batch_size=250,
+    use_variants=true,
+    Λ=[0.1, 0.2, 0.2],
+    nsamples=100,
+    niter_eccco=100
+)
\ No newline at end of file

experiments/jobscripts/counterfactuals/california_housing.sh

+#!/bin/bash
+
+#SBATCH --job-name="California Housing (ECCCo)"
+#SBATCH --time=3:00:00
+#SBATCH --ntasks=48
+#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. 
+
+module load 2023r1 openmpi
+
+srun julia --project=experiments experiments/run_experiments.jl -- data=california_housing output_path=results mpi > experiments/california_housing.log

experiments/jobscripts/counterfactuals/tabular.sh

 #!/bin/bash
 
 #SBATCH --job-name="Tabular (ECCCo)"
-#SBATCH --time=03:00:00
+#SBATCH --time=05:00:00
 #SBATCH --ntasks=48
 #SBATCH --cpus-per-task=1
 #SBATCH --partition=compute
@@ -11,4 +11,4 @@
 
 module load 2023r1 openmpi
 
-srun julia --project=experiments experiments/run_experiments.jl -- data=gmsc,german_credit,credit_default output_path=results mpi > experiments/tabular.log
+srun julia --project=experiments experiments/run_experiments.jl -- data=gmsc,german_credit,credit_default,california_housing output_path=results mpi > experiments/tabular.log

experiments/jobscripts/models/california_housing.sh

+#!/bin/bash
+
+#SBATCH --job-name="Train California Housing (ECCCo)"
+#SBATCH --time=01:00:00
+#SBATCH --ntasks=1
+#SBATCH --gpus-per-task=1
+#SBATCH --cpus-per-task=1
+#SBATCH --partition=gpu
+#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=california_housing output_path=results only_models > experiments/train_california_housing.log
\ No newline at end of file

experiments/run_experiments.jl

@@ -47,6 +47,12 @@ if "credit_default" in datanames
     include("credit_default.jl")
 end
 
+# California Housing
+if "california_housing" in datanames
+    @info "Running California Housing experiment."
+    include("california_housing.jl")
+end
+
 # MNIST
 if "mnist" in datanames
     @info "Running MNIST experiment."