if __name__ == "__main__": test_data_visualizer = False test_pre_processor = False test_dicom_file_manager = False test_file_manager = False test_classifier_comparator = True # ----------------------------------- FileManager ----------------------------------- if test_file_manager: fm = FileManager() training_path = 'covid19_repo/data/x_rays/covid' shape = (64, 64) shuffle = True x, y = fm.get_classifier_image_training_data( training_data_path=training_path, shape=shape, shuffle=shuffle) print(x.shape, y.shape) # ----------------------------------- DataVisualizer ----------------------------------- if test_data_visualizer: visualize_pca = False visualize_cnn_activations = True # ---------- Visualizing PCA --------- if visualize_pca: n_classes = 3 x = np.random.random((100, 32, 32)) y = np.random.randint(0, n_classes, size=100) dv = DataVisualizer(x=x, y=y)