|
@@ -20,7 +20,7 @@ perf_prediction_model_path = 'predictions_models_results.csv'
|
|
|
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"
|
|
|
perf_prediction_header_file = "data;data_size;model_name;accucary;f1;recall;precision;roc;\n"
|
|
|
|
|
|
-features_choices_labels = ['static', 'svd_reconstruction', 'fast_ica_reconstruction', 'ipca_reconstruction', 'min_diff_filter', 'sobel_based_filter']
|
|
|
+features_choices_labels = ['static', 'svd_reconstruction', 'fast_ica_reconstruction', 'ipca_reconstruction', 'min_diff_filter', 'sobel_based_filter','nl_mean_noise_mask']
|
|
|
|
|
|
# parameters
|
|
|
|