Dynamické přiřazování kategorií
Dynamické přiřazování kategorií umožňuje modelu zařadit text do předem definovaných kategorií, a to i bez předchozího tréninku na těchto kategoriích.
Pomocí funkce pipeline() z Hugging Face pro úlohu zero-shot-classification zadej text a předdefinované kategorie, aby model našel nejlepší shodu.
Sestaví klasifikátor, který předpoví štítek pro vstupní text – jde o novinový titulek, který je už načtený.
pipelines z knihovny transformers je předem načtený, takže ho můžeš rovnou použít.
Poznámka: Pracujeme s upravenou verzí pipeline, která ti umožní naučit se tyto funkce bez nutnosti stahovat model.
Toto cvičení je součástí kurzu
Working with Hugging Face
Pokyny k cvičení
- Sestav pipeline a ulož ji jako
classifier. - Vytvoř seznam štítků –
"politics","science","sports"– a ulož ho jakocategories. - Předpověz štítek pro
textpomocí klasifikátoru a předdefinovaných kategorií.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
text = "AI-powered robots assist in complex brain surgeries with precision."
# Create the pipeline
____ = pipeline(____="zero-shot-classification", ____="facebook/bart-large-mnli")
# Create the categories list
categories = ["politics", "____", "____"]
# Predict the output
output = ____(____, ____)
# Print the top label and its score
print(f"Top Label: {output['labels'][0]} with score: {output['scores'][0]}")