好像这个版本1.0.0后有的demo样例无法召回了, macbert4csc-base-chinese

#8
by Macropodus - opened

{'source': '机七学习是人工智能领遇最能体现智能的一个分知', 'target': '机器学习是人工智能领域最能体现智能的一个分知', 'errors': [('七', '器', 1), ('遇', '域', 10)]}
{'source': '一只小鱼船浮在平净的河面上', 'target': '一只小鱼船浮在平净的河面上', 'errors': []}
{'source': '他的语说的很好,法语也不错', 'target': '他的语说的很好,法语也不错', 'errors': []}

代码:

    model_name_or_path = "shibing624/macbert4csc-base-chinese"
    model_macbert = MacBertCorrector(model_name_or_path, device="cuda:0")

    texts = [
        '真麻烦你了。希望你们好好的跳无',
        '少先队员因该为老人让坐',
        '机七学习是人工智能领遇最能体现智能的一个分知',
        '一只小鱼船浮在平净的河面上',
        '我的家乡是有明的渔米之乡',
        '少先队员因该为老人让坐',
        '机七学习是人工智能领遇最能体现智能的一个分知',
        '今天新情很好',
        '老是较书。',
        '遇到一位很棒的奴生跟我聊天。',
        '他的语说的很好,法语也不错',
        '他法语说的很好,的语也不错',
        '他们的吵翻很不错,再说他们做的咖喱鸡也好吃',
        '影像小孩子想的快,学习管理的斑法',
        '餐厅的换经费产适合约会',
        '走路真的麻坊,我也没有喝的东西,在家汪了',
        '因为爸爸在看录音机,所以我没得看',
        '不过在许多传统国家,女人向未得到平等',
        '我想喝小明同学。',  # 漏召回
        '直接采用君主常用名,如赵昺。',  # 人名误杀
        '广东队其馀三场比赛均顺利拿下。',
        '检查范围复盖所有的在建保障性住房项目。',
        '禅精竭虑',
        '你找到你最喜欢的工作,我也很高心。',
        '今天新情很好',
        ]
    params = {
              "max_length": 128,
              "batch_size": 32,
              "threshold": 0.7,
              "silent": True
              }
    print("#"*128)
    for texts_i in texts:
        texts_i_predict = model_macbert.correct(texts_i, **params)
        print(texts_i_predict)

设置 "threshold": 0.1,

shibing624 changed discussion status to closed

Sign up or log in to comment