def get_settings(self): """ Initialize and customize settings. """ translation_settings = TranslationSettings() translation_settings.models = ["model.npz"] translation_settings.num_processes = 1 translation_settings.beam_width = 12 translation_settings.normalization_alpha = 1.0 translation_settings.suppress_unk = True return translation_settings
def get_settings(model, beam): """ Initialize and customize settings. """ translation_settings = TranslationSettings() translation_settings.models = [model] translation_settings.num_processes = 1 translation_settings.beam_width = beam translation_settings.normalization_alpha = 1.0 translation_settings.suppress_unk = True translation_settings.get_word_probs = False return translation_settings