好像这个版本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,
get
shibing624
changed discussion status to
closed