Fix one more

Yu-Group · Mar 19, 2024 · 377028f · 377028f
1 parent 96a0982
commit 377028f
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/feature_importance/01_ablation_classification_script.sh b/feature_importance/01_ablation_classification_script.sh
@@ -4,7 +4,7 @@
 seed=$(shuf -i 1-10000 -n 1)
 
 # Replace XXX with the generated seed
-command="01_run_ablation_classification.py --nreps 1 --config mdi_local.real_data_classification --split_seed $seed --ignore_cache --create_rmd --result_name try2.0"
+command="01_run_ablation_classification.py --nreps 1 --config mdi_local.real_data_classification --split_seed $seed --ignore_cache --create_rmd --result_name Diabetes_classification_parallel"
 
 # Execute the command
 python $command
diff --git a/feature_importance/01_ablation_regression_script.sh b/feature_importance/01_ablation_regression_script.sh
@@ -4,7 +4,7 @@
 seed=$(shuf -i 1-10000 -n 1)
 
 # Replace XXX with the generated seed
-command="01_run_ablation_regression.py --nreps 1 --config mdi_local.real_data_regression --split_seed $seed --ignore_cache --create_rmd --result_name try2.0"
+command="01_run_ablation_regression.py --nreps 1 --config mdi_local.real_data_regression --split_seed $seed --ignore_cache --create_rmd --result_name Diabetes_regression_parallel"
 
 # Execute the command
 python $command
diff --git a/feature_importance/01_ablation_script.sh b/feature_importance/01_ablation_script.sh
@@ -1,7 +1,9 @@
 #!/bin/bash
+#SBATCH [email protected]
+#SBATCH --mail-type=ALL
 
 # Define the SLURM submission script name
-slurm_script="{slurm_submission_script}.sh"  # Replace {slurm_submission_script} with your actual script name
+slurm_script="01_ablation_regression_script.sh"  # Replace {slurm_submission_script} with your actual script name
 
 # Loop to submit SLURM job 10 times
 for rep in {1..10}

diff --git a/feature_importance/01_run_ablation_regression.py b/feature_importance/01_run_ablation_regression.py
@@ -163,6 +163,7 @@ def compare_estimators(estimators: List[ModelConfig],
                         ablation_results_list[i] += mean_squared_error(y_test, est.predict(ablation_X_test))
                         ablation_results_list_r2[i] += r2_score(y_test, est.predict(ablation_X_test))
                 ablation_results_list = [x / len(seeds) for x in ablation_results_list]
+                ablation_results_list_r2 = [x / len(seeds) for x in ablation_results_list_r2]
                 for i in range(X_test.shape[1]):
                     metric_results[f'MSE_after_ablation_{i+1}'] = ablation_results_list[i]
                     metric_results[f'R_2_after_ablation_{i+1}'] = ablation_results_list_r2[i]