File size: 841 Bytes
c668e80 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
""" Modules for translation """
from onmt.translate.translator import Translator, GeneratorLM
from onmt.translate.translation import Translation, TranslationBuilder
from onmt.translate.beam_search import BeamSearch, GNMTGlobalScorer
from onmt.translate.beam_search import BeamSearchLM
from onmt.translate.decode_strategy import DecodeStrategy
from onmt.translate.greedy_search import GreedySearch, GreedySearchLM
from onmt.translate.penalties import PenaltyBuilder
from onmt.translate.translation_server import TranslationServer, ServerModelError
__all__ = [
"Translator",
"Translation",
"BeamSearch",
"GNMTGlobalScorer",
"TranslationBuilder",
"PenaltyBuilder",
"TranslationServer",
"ServerModelError",
"DecodeStrategy",
"GreedySearch",
"GreedySearchLM",
"BeamSearchLM",
"GeneratorLM",
]
|