diff --git a/results/breast_cancer/stage4_fix/losses.png b/results/breast_cancer/stage4_fix/losses.png
index 17a66350535125255ec2cfb56f672921967a3f33..577016a2ea1e40f9a026a7a7b47b839c98ae16da 100644
Binary files a/results/breast_cancer/stage4_fix/losses.png and b/results/breast_cancer/stage4_fix/losses.png differ
diff --git a/results/kr-vs-kp/stage4_fix/losses.png b/results/kr-vs-kp/stage4_fix/losses.png
index 4e1842193cfc9f1a4e3e9a7b75c2c30110c3c28f..af17ea8990d8de60ad2d9b544a533a42b68c0770 100644
Binary files a/results/kr-vs-kp/stage4_fix/losses.png and b/results/kr-vs-kp/stage4_fix/losses.png differ
diff --git a/results/steel-plates/stage4_fix/losses.png b/results/steel-plates/stage4_fix/losses.png
index 32f1ad27bbb83c693023323fd40ad03a096bd563..3a72cd7d2c0a674a272eb76b61295194faee64d5 100644
Binary files a/results/steel-plates/stage4_fix/losses.png and b/results/steel-plates/stage4_fix/losses.png differ
diff --git a/scripts/run_compute_results.sh b/scripts/run_compute_results.sh
index 3dd82f1fe9aeb2ad2bae066cc3709c0d61857010..74db5ec579d6df22cf7d5a8023d008253e3f629e 100644
--- a/scripts/run_compute_results.sh
+++ b/scripts/run_compute_results.sh
@@ -1,4 +1,4 @@
-for dataset in digits iris kin8nm kr-vs-kp lfw_pairs olivetti_faces wine 
+for dataset in steel-plates
 do
     python code/compute_results.py --stage=4 --experiment_ids 1 2 3 --dataset_name=$dataset --models_dir=models/$dataset/stage5
 done
diff --git a/scripts/run_stage5_experiments.sh b/scripts/run_stage5_experiments.sh
index a263555d40ebf6c695338584a95a6329702fdf7e..837231b6eb191b1651f955fe9408e77222655aca 100644
--- a/scripts/run_stage5_experiments.sh
+++ b/scripts/run_stage5_experiments.sh
@@ -4,7 +4,7 @@ walltime=1:00
 seeds='1 2 3 4 5 6 7'
 
 #for dataset in breast_cancer diabetes diamonds california_housing boston linnerud steel-plates
-for dataset in digits iris kin8nm kr-vs-kp lfw_pairs olivetti_faces wine
+for dataset in steel-plates
 do
     oarsub -p "(gpu is null)" -l /core=$core_number,walltime=1:00 "conda activate test_env && python code/train.py --dataset_name=$dataset --seeds $seeds --extraction_strategy=none --extracted_forest_size_stop=1 --extracted_forest_size_samples=30 --experiment_id=1 --models_dir=models/$dataset/stage5 --subsets_used train+dev,train+dev"
     oarsub -p "(gpu is null)" -l /core=$core_number,walltime=1:00 "conda activate test_env && python code/train.py --dataset_name=$dataset --seeds $seeds --extraction_strategy=random --extracted_forest_size_stop=1 --extracted_forest_size_samples=30 --experiment_id=2 --models_dir=models/$dataset/stage5 --subsets_used train+dev,train+dev"