diff --git a/results/boston/stage4_fix/losses.png b/results/boston/stage4_fix/losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..f33e126851096d963810b8ca7c93812eb0b314d7
Binary files /dev/null and b/results/boston/stage4_fix/losses.png differ
diff --git a/results/breast_cancer/stage4_fix/losses.png b/results/breast_cancer/stage4_fix/losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..17a66350535125255ec2cfb56f672921967a3f33
Binary files /dev/null and b/results/breast_cancer/stage4_fix/losses.png differ
diff --git a/results/california_housing/stage4_fix/losses.png b/results/california_housing/stage4_fix/losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..a43d3ff92aebddcf7d7fe02608fd52a38e2b0450
Binary files /dev/null and b/results/california_housing/stage4_fix/losses.png differ
diff --git a/results/diabetes/stage4_fix/losses.png b/results/diabetes/stage4_fix/losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..49249fce6b41fddd0c34dbe91e5891253722f58e
Binary files /dev/null and b/results/diabetes/stage4_fix/losses.png differ
diff --git a/results/diamonds/stage4_fix/losses.png b/results/diamonds/stage4_fix/losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..ac7725cc4ef10cab656037d6611a08d122b6cbee
Binary files /dev/null and b/results/diamonds/stage4_fix/losses.png differ
diff --git a/results/digits/stage4_fix/losses.png b/results/digits/stage4_fix/losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..dd0bd24c94e9239ee6eb0528e3a932d32df5e53e
Binary files /dev/null and b/results/digits/stage4_fix/losses.png differ
diff --git a/results/iris/stage4_fix/losses.png b/results/iris/stage4_fix/losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..386d06d375b78b29302a6516bc917a2f4c895437
Binary files /dev/null and b/results/iris/stage4_fix/losses.png differ
diff --git a/results/kin8nm/stage4_fix/losses.png b/results/kin8nm/stage4_fix/losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..82b96dc80d46d04fbfc7f56b4b18e3b999af2dc9
Binary files /dev/null and b/results/kin8nm/stage4_fix/losses.png differ
diff --git a/results/steel-plates/stage4_fix/losses.png b/results/steel-plates/stage4_fix/losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..32f1ad27bbb83c693023323fd40ad03a096bd563
Binary files /dev/null and b/results/steel-plates/stage4_fix/losses.png differ
diff --git a/results/wine/stage4_fix/losses.png b/results/wine/stage4_fix/losses.png
new file mode 100644
index 0000000000000000000000000000000000000000..4830942a79fb66938d166cc084145f1199de46a5
Binary files /dev/null and b/results/wine/stage4_fix/losses.png differ
diff --git a/scripts/run_compute_results.sh b/scripts/run_compute_results.sh
index d67571d78a9499b75a2c4558a517b01035025beb..41f1339ff630e1599c91297f2f76b2840a5d429d 100644
--- a/scripts/run_compute_results.sh
+++ b/scripts/run_compute_results.sh
@@ -1,5 +1,4 @@
-seeds='1 2 3'
-for dataset in boston iris diabetes digits linnerud wine breast_cancer olivetti_faces 20newsgroups_vectorized lfw_pairs california_housing diamonds
+for dataset in breast_cancer diabetes diamonds california_housing boston steel-plates digits iris kin8nm olivetti_faces wine
 do
-    python code/compute_results.py --stage=4 --experiment_ids 1 2 3 --dataset_name=$dataset --models_dir=models/$dataset/stage4
+    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 abd457632f655fbd92ff1b37a29d2a3edc929ed5..a263555d40ebf6c695338584a95a6329702fdf7e 100644
--- a/scripts/run_stage5_experiments.sh
+++ b/scripts/run_stage5_experiments.sh
@@ -3,8 +3,8 @@ core_number=5
 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 breast_cancer diabetes diamonds california_housing boston linnerud steel-plates
+for dataset in digits iris kin8nm kr-vs-kp lfw_pairs olivetti_faces wine
 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"