config.py 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. import os
  2. # Custom config used for redefined config variables if necessary
  3. # folders
  4. output_data_folder = 'data'
  5. output_data_generated = os.path.join(output_data_folder, 'generated')
  6. output_datasets = os.path.join(output_data_folder, 'datasets')
  7. output_zones_learned = os.path.join(output_data_folder, 'learned_zones')
  8. output_models = os.path.join(output_data_folder, 'saved_models')
  9. output_results_folder = os.path.join(output_data_folder, 'results')
  10. noisy_folder = 'noisy'
  11. not_noisy_folder = 'notNoisy'
  12. backup_model_folder = os.path.join(output_data_folder, 'models_backup')
  13. # file or extensions
  14. perf_prediction_model_path = 'predictions_models_results.csv'
  15. results_filename = 'results.csv'
  16. ## post_image_name_separator = '___'
  17. # variables
  18. perf_train_header_file = "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;\n"
  19. perf_prediction_header_file = "data;data_size;model_name;accucary;f1;recall;precision;roc;\n"
  20. features_choices_labels = ['static', 'svd_reconstruction', 'svd_reconstruction_dyn', 'fast_ica_reconstruction', 'ipca_reconstruction', 'min_diff_filter', 'sobel_based_filter','nl_mean_noise_mask', 'gini_map']
  21. # parameters
  22. sub_image_size = (200, 200)
  23. keras_epochs = 30
  24. ## keras_batch = 32
  25. ## val_dataset_size = 0.2
  26. keras_img_size = (200, 200)
  27. # parameters
  28. post_image_name_separator = '___'
  29. scene_image_quality_separator = '_'
  30. scene_image_extension = '.png'