load_tumor_image_data.py 文件源码

def __next__(self):
        if self._train_location >= len(self._train_indices)  or  self._test_location >= len(self._test_indices):
            # Reset:
            np.shuffle(self._train_indices)
            np.shuffle(self._test_indices)
            self._train_location = 0
            self._test_location  = 0
            raise StopIteration()
        return self.get_train_test_batch()