关键词快速匹配范例

   在进行文本分类时,仅仅依靠模型是得到满意的效果,需要结合关键词提高精度,一个例子如下:

复制代码
df = pd.read_excel(file_name, encoding='utf8')
text = df.values.tolist()
# key_words = ['牛肉','猪肉','鸡肉']
key_words = ['野鸡','野兔', '野生动物']
res = []
with open('./data/data_2020.txt', 'w', encoding='utf8') as fw:
    for i, each in enumerate(text):
        data = []
        for c in each:
            if isinstance(c, str):
                data.append(c)
        to_write = data[0]
        if len(to_write) >= 5 and len(to_write) <= 150:
            # 如果里面包括关键词:
            if any((to_write.find(k) != -1) for k in key_words):
                print(to_write)
                res.append(to_write.strip()+'\n')

    fw.writelines(res)
复制代码

 

posted @   今夜无风  阅读(415)  评论(0编辑  收藏  举报
编辑推荐:
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 理解Rust引用及其生命周期标识(上)
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析
· 没有源码,如何修改代码逻辑?
阅读排行:
· 分享4款.NET开源、免费、实用的商城系统
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
· 上周热点回顾(2.24-3.2)
点击右上角即可分享
微信分享提示