You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.0 KiB
43 lines
1.0 KiB
# -*- coding: utf-8 -*-
|
|
"""
|
|
@Time : 2021-02-03 21:57:15
|
|
@File : correct_demo.py
|
|
@Author : Abtion
|
|
@Email : abtion{at}outlook.com
|
|
"""
|
|
import argparse
|
|
import sys
|
|
|
|
from pycorrector.macbert.macbert_corrector import MacBertCorrector
|
|
from pycorrector import config
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
|
|
# Required parameters
|
|
parser.add_argument("--macbert_model_dir", default=config.macbert_model_dir,
|
|
type=str,
|
|
help="MacBert pre-trained model dir")
|
|
args = parser.parse_args()
|
|
|
|
nlp = MacBertCorrector(args.macbert_model_dir).macbert_correct
|
|
|
|
i = nlp('今天新情很好')
|
|
print(i)
|
|
|
|
i = nlp('少先队员英该为老人让座')
|
|
print(i)
|
|
|
|
i = nlp('机器学习是人工智能领遇最能体现智能的一个分知。')
|
|
print(i)
|
|
|
|
i = nlp('机其学习是人工智能领遇最能体现智能的一个分知。')
|
|
print(i)
|
|
|
|
print(nlp('老是较书。'))
|
|
print(nlp('遇到一位很棒的奴生跟我聊天。'))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|