Files
bertalign/bin/bleualign/example/default.py
2021-05-17 23:44:19 +08:00

32 lines
1.3 KiB
Python

from __future__ import print_function
import os
from bleualign.align import Aligner
if __name__ == '__main__':
current_path = os.path.dirname(os.path.abspath(__file__))
options = {
# source and target files needed by Aligner
# they can be filenames, arrays of strings or io objects.
'srcfile':os.path.join(current_path, '..', 'eval', 'eval1989.de'),
'targetfile': os.path.join(current_path, '..', 'eval', 'eval1989.fr'),
# translations of srcfile and targetfile, not influenced by 'factored'
# they can be filenames, arrays of strings or io objects, too.
'srctotarget': [os.path.join(current_path, '..', 'eval', 'eval1957.europarlfull.fr')],
'targettosrc': [],
# passing filenames or io object for them in respectly.
# if not passing anything or assigning None, they will use StringIO to save results.
'output-src': None, 'output-target': None,
# other options ...
}
a = Aligner(options)
a.mainloop()
output_src, output_target = a.results()
# output_src, output_target is StringIO because options['output-src'] is None
src = output_src.getvalue() # StringIO member function
trg = output_target.getvalue().splitlines() # array of string
print('output_src.getvalue()')
print(src[:30])
print()
print('output_target.getvalue().splitlines()')
print(trg[:3])