Feature/data augmentation

[vidvath7]

chebi_augmentation.yml

aug_data:
Set this parameter to True to enable the generation of an augmented dataset. When set to False, only the regular dataset will be generated.

augment_data_batch_size:
Configure this parameter to define the batch size for processing the augmented data.pkl file.

num_smiles_variations :
Use this parameter to specify the maximum number of SMILES variations to generate for each compound. This helps control the diversity of the generated dataset.

chebi.py:

AugmentedDataExtractor Class
This class inherits from the main ChEBI data extractor to specialize in generating augmented datasets. It supports custom configurations such as batch size and number of SMILES variations.

augment_data():
Verifies the existence of the original data.pkl file and, if found, generates and saves the augmented data.pkl file in the specified augmented directory.

generate_smiles_variations():
Produces SMILES variations based on different configurations like rooted atoms and randomization.

setup_processed():
Prepares processed data for the augmented ChEBI dataset by transforming and saving it in the required format.