File size: 854 Bytes
1484d4b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
def XlitEngine(
lang2use = "all", beam_width=4, rescore=True,
model_type = "transformer",
src_script_type = "roman",
):
if model_type == "transformer":
if src_script_type in {"roman", "latin", "en"}:
from .transformer import XlitEngineTransformer_En2Indic
return XlitEngineTransformer_En2Indic(lang2use, beam_width=beam_width, rescore=rescore)
elif src_script_type == "indic":
from .transformer import XlitEngineTransformer_Indic2En
return XlitEngineTransformer_Indic2En(beam_width=beam_width, rescore=rescore)
elif model_type == "rnn":
assert src_script_type in {"roman", "latin", "en"}
from .rnn.engine import XlitEngineRNN
return XlitEngineRNN(lang2use, beam_width=beam_width, rescore=rescore)
raise NotImplementedError()
|