custom_config.py 1.1 KB

1234567891011121314151617181920212223242526272829303132
  1. from modules.config.cnn_config import *
  2. # store all variables from cnn config
  3. context_vars = vars()
  4. # Custom config used for redefined config variables if necessary
  5. # folders
  6. ## noisy_folder = 'noisy'
  7. ## not_noisy_folder = 'notNoisy'
  8. backup_model_folder = 'models_backup'
  9. # file or extensions
  10. perf_prediction_model_path = 'predictions_models_results.csv'
  11. ## post_image_name_separator = '___'
  12. # variables
  13. 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"
  14. perf_prediction_header_file = "data;data_size;model_name;accucary;f1;recall;precision;roc;\n"
  15. features_choices_labels = ['static', 'svd_reconstruction', 'fast_ica_reconstruction', 'ipca_reconstruction', 'min_diff_filter']
  16. # parameters
  17. sub_image_size = (200, 200)
  18. keras_epochs = 30
  19. ## keras_batch = 32
  20. ## val_dataset_size = 0.2
  21. keras_img_size = (100, 100)