SinCode / sincode_model.py
KalanaPabasara
SinCode v3 — seq2seq pipeline, evaluation scripts, IndoNLP benchmark data
1fed70a
raw
history blame contribute delete
659 Bytes
"""
SinCode v3 — public API entry point.
Usage:
from sincode_model import BeamSearchDecoder
decoder = BeamSearchDecoder()
result, logs, word_candidates = decoder.decode("mema videowe bit rate eka godak wadi nisa buffer wenawa")
"""
from core.decoder import BeamSearchDecoder, ScoredCandidate # noqa: F401
from core.english import ENGLISH_VOCAB # noqa: F401
from core.constants import ( # noqa: F401
DEFAULT_MLM_MODEL, DEFAULT_BYT5_MODEL, DEFAULT_MBART_MODEL,
MAX_CANDIDATES, MIN_ENGLISH_LEN,
)
from seq2seq.mbart_infer import SentenceTransliterator # noqa: F401