Пример #1
0
 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()
Пример #2
0
 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()