--- license: mit datasets: - notaphoenix/shakespeare_dataset language: - en metrics: - f1 pipeline_tag: text-classification --- # Shakespeare/Modern English DistilBert-base # Description ℹ With this model, you can classify if an English sentence has a *Shakespearean* style or a *modern* style The model is a fine-tuned checkpoint of [DistilBERT-base-uncased](https://huggingface.co/distilbert-base-uncased). # Application 🚀 ## Return all labels ```python from transformers import pipeline classifier = pipeline("text-classification", model="notaphoenix/shakespeare_classifier_model", top_k=None) classifier("This is a modern sentence!") ``` ```json [[ {'label': 'modern', 'score': 0.901931643486023}, {'label': 'shakespearean', 'score': 0.09806833416223526} ]] ``` ## Return top label ```python from transformers import pipeline classifier = pipeline("text-classification", model="notaphoenix/shakespeare_classifier_model") classifier("This is a modern sentence!") ``` ```json [ {'label': 'modern', 'score': 0.901931643486023} ] ```