12345678910111213141516171819202122232425262728293031323334353637383940414243444546 |
- # -*- coding: utf-8 -*-
- """
- @author:XuMing(xuming624@qq.com)
- @description:
- """
- import argparse
- import sys
- sys.path.append('../..')
- from pycorrector.deepcontext.deepcontext_corrector import DeepContextCorrector
- def main():
- parser = argparse.ArgumentParser()
- parser.add_argument("--output_dir", default="./outputs-deepcontext-lm", type=str, help="Dir for model save.")
- parser.add_argument("--max_length", default=1024, type=int, help="The maximum total input sequence length")
- args = parser.parse_args()
- print(args)
- m = DeepContextCorrector(args.output_dir, max_length=args.max_length)
- inputs = [
- '老是较书。',
- '感谢等五分以后,碰到一位很棒的奴生跟我可聊。',
- '遇到一位很棒的奴生跟我聊天。',
- '遇到一位很美的女生跟我疗天。',
- '他们只能有两个选择:接受降新或自动离职。',
- '王天华开心得一直说话。'
- ]
- for i in inputs:
- output = m.correct(i)
- print('input :', i)
- print('predict:', output)
- print()
- # result:
- # input : 老是较书。
- # predict: 老师教书。 [('是', '师', 1, 2), ('较', '教', 2, 3)]
- #
- # input : 感谢等五分以后,碰到一位很棒的奴生跟我可聊。
- # predict: 感谢等五分以后,碰到一位很棒的女生跟我可聊。 [('奴', '女', 15, 16)]
- if __name__ == "__main__":
- main()
|