1. Learn
    2. /
  2. 课程
    3. /
  3. 使用 Hugging Face
Connected

道练习

动态类别分配

动态类别分配使模型即使未针对某些类别进行过训练，也能将文本归入预定义类别。

使用 Hugging Face 的 pipeline() 并选择 zero-shot-classification 任务，提供文本和预定义类别来找出最佳匹配。

构建一个分类器来预测输入 text 的标签。该输入是一条已为您加载的新闻标题。

已为您预加载 transformers 库中的 pipelines 以便使用。

注意： 我们使用了自定义版本的 pipeline，帮助您在无需下载模型的情况下学习如何使用这些函数。

说明

100 XP
  • 构建 pipeline 并保存为 classifier。
  • 创建标签列表："politics"、"science"、"sports"，并保存为 categories。
  • 使用分类器和预定义类别预测 text 的标签。