1. Learn
  2. /
  3. Курси
  4. /
  5. Робота з Hugging Face

Connected

вправа

Динамічне призначення категорій

Динамічне призначення категорій дає змогу моделі класифікувати текст за наперед визначеними категоріями, навіть без попереднього навчання на цих категоріях.

Використовуючи pipeline() від Hugging Face для завдання zero-shot-classification, передайте текст і наперед визначені категорії, щоб знайти найкращу відповідність.

Створіть класифікатор, який передбачить мітку для вхідного text — це заголовок новини, уже завантажений для вас.

pipelines із бібліотеки transformers попередньо імпортовано для вашої зручності.

Примітка: Ми використовуємо спеціально налаштовану версію pipeline, щоб ви могли навчитися користуватися цими функціями без потреби завантажувати модель.

Інструкції

100 XP
  • Побудуйте pipeline і збережіть його як classifier.
  • Створіть список міток — "politics", "science", "sports" — і збережіть його як categories.
  • Передбачте мітку для text, використавши класифікатор і наперед визначені категорії.