run run run

experiments/experiment.jl

@@ -54,11 +54,11 @@ end
 
 Train the models specified by `exper` and store them in `outcome`.
 """
-function train_models!(outcome::ExperimentOutcome, exper::Experiment)
+function train_models!(outcome::ExperimentOutcome, exper::Experiment; save_meta::Bool=false)
     model_dict = prepare_models(exper)
     outcome.model_dict = model_dict
     if !(is_multi_processed(exper) && MPI.Comm_rank(exper.parallelizer.comm) != 0)
-        meta_model_performance(outcome)
+        meta_model_performance(outcome; save_output=save_meta)
     end
 end
 
@@ -96,9 +96,12 @@ function run_experiment(exper::Experiment; save_output::Bool=true, only_models::
     end
 
     # Model training:
-    train_models!(outcome, exper)
-    # Return if only models are needed:
-    !only_models || return outcome
+    if only_models
+        train_models!(outcome, exper; save_meta=true)
+        return outcome
+    else
+        train_models!(outcome, exper)
+    end
 
     # Benchmark:
     benchmark!(outcome, exper)

experiments/jobscripts/tuning/generators/innovation/synthetic.sh

+#!/bin/bash
+
+#SBATCH --job-name="Grid-search Synthetic (ECCCo)"
+#SBATCH --time=06: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=linearly_separable,moons,circles output_path=results mpi grid_search > experiments/grid_search_synthetic.log

experiments/jobscripts/tuning/generators/innovation/tabular.sh

+#!/bin/bash
+
+#SBATCH --job-name="Grid-search Tabular (ECCCo)"
+#SBATCH --time=16: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=gmsc,german_credit,credit_default,california_housing output_path=results mpi grid_search > experiments/grid_search_tabular.log
\ No newline at end of file

experiments/jobscripts/tuning/generators/synthetic.sh

@@ -11,4 +11,4 @@
 
 module load 2023r1 openmpi
 
-srun julia --project=experiments experiments/run_experiments.jl -- data=linearly_separable,moons,circles output_path=results mpi grid_search > experiments/synthetic.log
+srun julia --project=experiments experiments/run_experiments.jl -- data=linearly_separable,moons,circles output_path=results mpi grid_search > experiments/grid_search_synthetic.log

experiments/jobscripts/tuning/generators/tabular.sh

@@ -11,4 +11,4 @@
 
 module load 2023r1 openmpi
 
-srun julia --project=experiments experiments/run_experiments.jl -- data=gmsc,german_credit,credit_default,california_housing output_path=results mpi grid_search > experiments/synthetic.log
+srun julia --project=experiments experiments/run_experiments.jl -- data=gmsc,german_credit,credit_default,california_housing output_path=results mpi grid_search > experiments/grid_search_tabular.log
\ No newline at end of file