Skip to content
Snippets Groups Projects
Commit bb57ae1b authored by Pat Alt's avatar Pat Alt
Browse files

more job scripts

parent fd438e4b
No related branches found
No related tags found
1 merge request!7669 initial run including fmnist lenet and new method
counterfactual_data, test_data = train_test_split(load_credit_default(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
#!/bin/bash
#SBATCH --job-name="Credit Default (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=credit_default output_path=results mpi > experiments/credit_default.log
#!/bin/bash
#SBATCH --job-name="Tabular (ECCCo)"
#SBATCH --time=03: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 output_path=results mpi > experiments/tabular.log
#!/bin/bash
#SBATCH --job-name="Train Credit Default (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=credit_default output_path=results only_models > experiments/train_credit_default.log
\ No newline at end of file
......@@ -10,4 +10,4 @@
#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
srun julia --project=experiments experiments/run_experiments.jl -- data=german_credit output_path=results only_models > experiments/train_german_credit.log
\ No newline at end of file
......@@ -10,4 +10,4 @@
#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
srun julia --project=experiments experiments/run_experiments.jl -- data=mnist output_path=results only_models > experiments/train_mnist.log
\ No newline at end of file
......@@ -41,6 +41,12 @@ if "german_credit" in datanames
include("german_credit.jl")
end
# Credit Default
if "credit_default" in datanames
@info "Running Credit Default experiment."
include("credit_default.jl")
end
# MNIST
if "mnist" in datanames
@info "Running MNIST experiment."
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment