huggingface pipeline零训练样本分类Zero-Shot Classification的实现

hehui0921

已于 2024-02-13 17:05:56 修改

阅读量686

点赞数 1

分类专栏： huggingface 文章标签：分类 python 数据挖掘

于 2024-02-12 11:57:42 首次发布

本文链接：https://blog.csdn.net/oHeHui1/article/details/136100369

版权

1 ：默认的model 。

from huggingface_hub.hf_api import HfFolder
HfFolder.save_token('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')
from transformers import MBartForConditionalGeneration, MBart50TokenizerFast

from transformers import pipeline

classifier = pipeline("zero-shot-classification")
result = classifier(
"This is a course about the Transformers library",
candidate_labels=["education",