def __init__(self, train_examples, test_examples=None, add_generated_examples=True):
"""Initialize the dataset using the provided train and test examples."""
self.logger.info("Loading DS from files...")
self.augmenter = SignalAugmenter(augmentation_start=0.1, augmentation_end=0.9)
train = self.prepare_dataset(train_examples, add_generated_examples)
test = self.prepare_dataset(test_examples, add_generated_examples)
self.id_label_mapping = {v: k for k, v in self.label_id_mapping.items()}
self.X_train = self.flatten2d(train.features)
self.y_train = train.labels
self.X_test = self.flatten2d(test.features)
self.y_test = test.labels
self.num_labels = len(self.id_label_mapping)
self.num_features = self.X_train.shape[1]
self.num_train_examples = self.X_train.shape[0]
self.num_test_examples = self.X_test.shape[0]
