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)
