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]