def test_generator(self): self._reset() _, test_dataset_dictionaries_mask = self.dataset.get_test_dataset() while True: test_dataset_dictionaries_mask = shuffle_dict( test_dataset_dictionaries_mask) yield from self.preprocess_batch(test_dataset_dictionaries_mask) self._reset()
def validation_generator(self): self._reset() _, validation_dataset_dictionaries_mask = self.dataset.get_validation_dataset( ) while True: validation_dataset_dictionaries_mask = shuffle_dict( validation_dataset_dictionaries_mask) yield from self.preprocess_batch( validation_dataset_dictionaries_mask) self._reset()