def preprocess_data(args): """ Function to process the data and create data sets. :param args: command line arguments """ # Create directories for font and plots if they do not exist if not os.path.exists(args.font_dir): os.mkdir(args.font_dir) if not os.path.exists(args.plots_dir): os.mkdir(args.plots_dir) # Create objects for preprocessing and visualization data_preprocessor = DataPreprocessor(args) visualizer = Visualizer(args.font_dir, args.plots_dir) # Get information from DEAM files data_preprocessor.get_data_info() # Get waveforms from audio data_preprocessor.get_waveforms() # Augment dataset data_preprocessor.augment_quadrants() # Create sets for train and test data_preprocessor.make_train_test_sets() # Visualize data distribution in quadrants visualizer.visualize_data_distribution(data_preprocessor.annotations, data_preprocessor.quadrants) # Visualize data distribution on each valence and arousal dimensions visualizer.visualize_dimensions_distribution(data_preprocessor.annotations)