let's do this

experiments/california_housing.jl

@@ -3,6 +3,9 @@ dataname = "California Housing"
 counterfactual_data, test_data =
     train_test_split(load_california_housing(nothing); test_size = TEST_SIZE)
 
+# Domain constraints:
+counterfactual_data.domain = extrema(counterfactual_data.X, dims=2)
+
 # VAE:
 using CounterfactualExplanations.GenerativeModels: VAE, train!
 X = counterfactual_data.X

experiments/circles.jl

@@ -4,6 +4,9 @@ n_obs = Int(1000 / (1.0 - TEST_SIZE))
 counterfactual_data, test_data =
     train_test_split(load_circles(n_obs; noise = 0.05, factor = 0.5); test_size = TEST_SIZE)
 
+# Domain constraints:
+counterfactual_data.domain = extrema(counterfactual_data.X, dims=2)
+
 # Model tuning:
 model_tuning_params = DEFAULT_MODEL_TUNING_SMALL
 

experiments/daic/testing/cali_100_8.sh

 #!/bin/bash
 
 #SBATCH --job-name="Grid-search California Housing (ECCCo)"
-#SBATCH --time=01:10:00
-#SBATCH --ntasks=10
-#SBATCH --cpus-per-task=10
+#SBATCH --time=00:35:00
+#SBATCH --ntasks=14
+#SBATCH --cpus-per-task=14
 #SBATCH --partition=general
 #SBATCH --mem-per-cpu=2GB
 #SBATCH --mail-type=END     # Set mail type to 'END' to receive a mail when the job finishes. 

experiments/german_credit.jl

@@ -3,6 +3,9 @@ dataname = "German Credit"
 counterfactual_data, test_data =
     train_test_split(load_german_credit(nothing); test_size = TEST_SIZE)
 
+# Domain constraints:
+counterfactual_data.domain = extrema(counterfactual_data.X, dims=2)
+
 # VAE:
 using CounterfactualExplanations.GenerativeModels: VAE, train!
 X = counterfactual_data.X

experiments/gmsc.jl

@@ -3,6 +3,9 @@ dataname = "GMSC"
 counterfactual_data, test_data = train_test_split(load_gmsc(nothing); test_size = TEST_SIZE)
 nobs = size(counterfactual_data.X, 2)
 
+# Domain constraints:
+counterfactual_data.domain = extrema(counterfactual_data.X, dims=2)
+
 # VAE:
 using CounterfactualExplanations.GenerativeModels: VAE, train!
 X = counterfactual_data.X

experiments/jobscripts/testing/cali.sh

+#!/bin/bash
+
+#SBATCH --job-name="Grid-search California Housing (ECCCo)"
+#SBATCH --time=00:35:00
+#SBATCH --ntasks=14
+#SBATCH --cpus-per-task=14
+#SBATCH --partition=compute
+#SBATCH --mem-per-cpu=2GB
+#SBATCH --account=research-eemcs-insy
+#SBATCH --mail-type=END     # Set mail type to 'END' to receive a mail when the job finishes. 
+
+module use /opt/insy/modulefiles          # Use DAIC INSY software collection
+module load openmpi
+
+source experiments/slurm_header.sh
+
+srun julia --project=experiments --threads $SLURM_CPUS_PER_TASK experiments/run_experiments.jl -- data=california_housing output_path=results mpi grid_search n_individuals=10 threaded > experiments/grid_search_california_housing.log
\ No newline at end of file

experiments/linearly_separable.jl

@@ -6,6 +6,9 @@ counterfactual_data, test_data = train_test_split(
     test_size = TEST_SIZE,
 )
 
+# Domain constraints:
+counterfactual_data.domain = extrema(counterfactual_data.X, dims=2)
+
 # Model tuning:
 model_tuning_params = DEFAULT_MODEL_TUNING_SMALL
 

experiments/moons.jl

@@ -3,6 +3,9 @@ dataname = "Moons"
 n_obs = Int(2500 / (1.0 - TEST_SIZE))
 counterfactual_data, test_data = train_test_split(load_moons(n_obs); test_size = TEST_SIZE)
 
+# Domain constraints:
+counterfactual_data.domain = extrema(counterfactual_data.X, dims=2)
+
 # Model tuning:
 model_tuning_params = DEFAULT_MODEL_TUNING_SMALL