txtai 教程系列 7-----应用具有零样本分类的标签

本文介绍如何使用txtai的Labels实例进行零样本分类,以文本为输入,选择最匹配的标签。示例中展示了如何将标签应用于体育赛事相关的文本片段,选择正确的运动类型。此外,还探讨了模型对表情符号的理解。
摘要由CSDN通过智能技术生成

本文展示了如何使用零样本分类来执行文本分类、标记和主题建模。txtai 围绕 Hugging Face Transformers 中的零镜头分类管道提供了一个轻量级包装器。这种方法开箱即用,效果非常好。感谢 Hugging Face 团队在零样本分类方面的出色工作!

本文中的示例使用文本片段的标签列表来选择最匹配的标签。

tldrstory使用 Streamlit、FastAPI 和 Hugging Face Transformers 实现了零样本分类系统的全栈实现。
安装依赖
安装txtai和所有依赖项。
pip install txtai
创建标签实例
Labels 实例是零样本分类的主要入口点。这是围绕 Hugging Face Transformers 中的零镜头分类管道的轻量级包装器。

除了默认模型外,还可以在Hugging Face 模型中心找到其他模型。
from txtai.pipeline import Labels

Create labels model

labels = Labels()
将标签应用于文本
下面的示例显示了如何将零样本分类器应用于任意文本。零样本分类管道的默认模型是bart-large-mnli。

看看下面的结果。它的表现简直令人惊叹✨。即使对人类来说,这些也不简单。例如,故意选择拦截,因为这在足球中比在篮球中更常见。更大的 Transformer 模型中存储的知识量继续给我留下深刻印象。
data = [“Dodgers lose again, give up 3 HRs in a loss to the Giants”,
“Giants 5 Cardinals 4 final in extra innings”,
“Dodgers drop Game 2 against the Giants, 5-4”,
“Fl

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Q shen

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值