fancyzhx/ag_news
Viewer • Updated • 128k • 81.1k • 194
How to use joeddav/distilbert-base-uncased-agnews-student with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="joeddav/distilbert-base-uncased-agnews-student") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("joeddav/distilbert-base-uncased-agnews-student")
model = AutoModelForSequenceClassification.from_pretrained("joeddav/distilbert-base-uncased-agnews-student", device_map="auto")This model is distilled from the zero-shot classification pipeline on the unlabeled AG's News dataset using this script. It is the result of the demo notebook here, where more details about the model can be found.
"This text is about {}."The model can be used like any other model trained on AG's News, but will likely not perform as well as a model trained with full supervision. It is primarily intended as a demo of how an expensive NLI-based zero-shot model can be distilled to a more efficient student.