run.sh 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. #!/bin/bash
  2. erased=$1
  3. if [ "${erased}" == "Y" ]; then
  4. echo "Previous data file erased..."
  5. rm ${file_path}
  6. mkdir -p models_info
  7. touch ${file_path}
  8. # add of header
  9. echo 'model_name; global_train_size; global_test_size; filtered_train_size; filtered_test_size; f1_train; f1_test; recall_train; recall_test; presicion_train; precision_test; acc_train; acc_test; roc_auc_train; roc_auc_test;' >> ${file_path}
  10. fi
  11. metric="svd_reconstruction"
  12. # file which contains model names we want to use for simulation
  13. comparisons_models="models_info/models_comparisons.csv"
  14. for begin in {80,85,90,95,100,105,110}; do
  15. for end in {150,160,170,180,190,200}; do
  16. # python generate_reconstructed_data.py --metric ${metric} --interval "${begin}, ${end}"
  17. for zone in {6,8,10,12}; do
  18. OUTPUT_DATA_FILE="${metric}_nb_zones_${zone}_B${begin}_E${end}"
  19. if grep -xq "${OUTPUT_DATA_FILE}" "${comparisons_models}"; then
  20. echo "Run simulation for model ${OUTPUT_DATA_FILE}"
  21. python generate_dataset.py --output data/${OUTPUT_DATA_FILE} --metric ${metric} --renderer "maxwell" --scenes "A, D, G, H" --interval "${begin}, ${end}" --nb_zones ${zone} --random 1
  22. python train_model.py --data data/${OUTPUT_DATA_FILE} --output ${OUTPUT_DATA_FILE}
  23. done
  24. done
  25. done