| """ 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"] | |