|
|
@ -171,8 +171,9 @@ def train(src_dict_size, trg_dict_size, src_lang="en"):
|
|
|
|
callable: The train reader.
|
|
|
|
callable: The train reader.
|
|
|
|
"""
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
|
|
assert (src_lang in ["en", "de"], ("An error language type. Only support: "
|
|
|
|
if src_lang not in ["en", "de"]:
|
|
|
|
"en (for English); de(for Germany)"))
|
|
|
|
raise ValueError("An error language type. Only support: "
|
|
|
|
|
|
|
|
"en (for English); de(for Germany).")
|
|
|
|
src_dict_size, trg_dict_size = __get_dict_size(src_dict_size, trg_dict_size,
|
|
|
|
src_dict_size, trg_dict_size = __get_dict_size(src_dict_size, trg_dict_size,
|
|
|
|
src_lang)
|
|
|
|
src_lang)
|
|
|
|
|
|
|
|
|
|
|
@ -218,9 +219,9 @@ def test(src_dict_size, trg_dict_size, src_lang="en"):
|
|
|
|
callable: The test reader.
|
|
|
|
callable: The test reader.
|
|
|
|
"""
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
|
|
assert (src_lang in ["en", "de"],
|
|
|
|
if src_lang not in ["en", "de"]:
|
|
|
|
("An error language type. "
|
|
|
|
raise ValueError("An error language type. "
|
|
|
|
"Only support: en (for English); de(for Germany)"))
|
|
|
|
"Only support: en (for English); de(for Germany).")
|
|
|
|
|
|
|
|
|
|
|
|
src_dict_size, trg_dict_size = __get_dict_size(src_dict_size, trg_dict_size,
|
|
|
|
src_dict_size, trg_dict_size = __get_dict_size(src_dict_size, trg_dict_size,
|
|
|
|
src_lang)
|
|
|
|
src_lang)
|
|
|
@ -266,9 +267,9 @@ def validation(src_dict_size, trg_dict_size, src_lang="en"):
|
|
|
|
Returns:
|
|
|
|
Returns:
|
|
|
|
callable: The validation reader.
|
|
|
|
callable: The validation reader.
|
|
|
|
"""
|
|
|
|
"""
|
|
|
|
assert (src_lang in ["en", "de"],
|
|
|
|
if src_lang not in ["en", "de"]:
|
|
|
|
("An error language type. "
|
|
|
|
raise ValueError("An error language type. "
|
|
|
|
"Only support: en (for English); de(for Germany)"))
|
|
|
|
"Only support: en (for English); de(for Germany).")
|
|
|
|
src_dict_size, trg_dict_size = __get_dict_size(src_dict_size, trg_dict_size,
|
|
|
|
src_dict_size, trg_dict_size = __get_dict_size(src_dict_size, trg_dict_size,
|
|
|
|
src_lang)
|
|
|
|
src_lang)
|
|
|
|
|
|
|
|
|
|
|
|