Commit fa243135 authored by szr712's avatar szr712

测试政史地模型第一版

parent 3f931405
......@@ -65,7 +65,7 @@ def cer(preFile,textFile):
round(float(all_edit_distance)/all_chars, 5)))
if __name__ == "__main__":
preFile = "./data/test_data/end2end/Chinese/hanzi"
textFile = "./data/test_data/end2end/gt"
preFile = "./data/test_data/end2end_zsd/pre"
textFile = "./data/test_data/end2end_zsd/gt"
cer(preFile,textFile)
\ No newline at end of file
......@@ -47,3 +47,6 @@ CUDA_VISIBLE_DEVICES=5 python train_token_classification.py -src_data data/train
CUDA_VISIBLE_DEVICES=4 python translate2.py -load_weights weights/token_classification_aug/token_classification_split_new_119_0.0013118577939167154 -pkl_dir weights/token_classification_aug -src_voc ./data/voc/pinyin.txt -trg_voc ./data/voc/hanzi.txt
CUDA_VISIBLE_DEVICES=4 python translate_file2.py -load_weights weights/token_classifiaction_aug_Chinese/token_classification_split_Chinese_148_0.012723878987599165 -pkl_dir weights/token_classifiaction_aug_Chinese -src_voc ./data/voc/pinyin.txt -trg_voc ./data/voc/hanzi.txt -test_dir data/test_data/end2end/Chinese/pinyin -result_dir data/test_data/end2end/Chinese/hanzi
CUDA_VISIBLE_DEVICES=4 python translate_file2.py -load_weights weights/token_classification_split_zsd/token_classification_split_zsd/token_classification_split_zsd_98_0.0004903546059381369 -pkl_dir weights/token_classification_split_zsd/token_classification_split_zsd -src_voc ./data/voc/pinyin.txt -trg_voc ./data/voc/hanzi.txt -test_dir data/test_data/end2end_zsd/pinyin -result_dir data/test_data/end2end_zsd/pre
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment