model.py 文件源码

def predict_from_dataset(self, dataset: Dataset,
                             show_eos: bool=True,
                             use_queue: bool=True,
                             **kwargs) -> Iterator[Tuple[str, str, str]]:
        if use_queue:
            evaluator = self._predict_from_dataset_queue(dataset, **kwargs)
        else:
            evaluator = self._predict_from_dataset_feed(dataset, **kwargs)

        for source, target, translation in evaluator:

            # unpack and decode result
            yield from zip(
                self.dataset.decode_as_batch(source,
                                             show_eos=show_eos),
                self.dataset.decode_as_batch(target,
                                             show_eos=show_eos),
                self.dataset.decode_as_batch(translation,
                                             show_eos=show_eos)
            )