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
def get_settings(self):
"""
Initialize and customize settings.
"""
decoder_settings = DecoderSettings()
decoder_settings.models = ["model.npz"]
decoder_settings.num_processes = 1
translation_settings = TranslationSettings()
translation_settings.beam_width = 12
translation_settings.normalization_alpha = 1.0
translation_settings.suppress_unk = True
translation_settings.get_word_probs = True
return decoder_settings, translation_settings
def outputEqual(output1, output2):
"""given two translation outputs, check that output string is identical
"""
for i, (line, line2) in enumerate(
zip(open(output1).readlines(),
open(output2).readlines())):
#assertEqual(line.strip(), line2.strip())
print "translate {}".format(i)
print line
print line2
# English-German WMT16 system, no dropout
"""
Initialize and customize settings.
"""
translation_settings = TranslationSettings()
translation_settings.models = ["model_test/model.npz-80000"]
#translation_settings.num_processes = 8
translation_settings.beam_width = 10
translation_settings.normalization_alpha = 1.0
translation_settings.verbose = True
translation_settings.n_best = True
translation_settings.suppress_unk = True
translate(input_file=open('data/translated/fr.00'),
output_file=open('data/fr.00.8.out', 'w'),
translation_settings=translation_settings)
print "everyting ok"
#outputEqual('en-de/ref2','en-de/out')
