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.
Application 🚀
Return all labels
from transformers import pipeline
classifier = pipeline("text-classification", model="notaphoenix/shakespeare_classifier_model", top_k=None)
classifier("This is a modern sentence!")
[[
{'label': 'modern', 'score': 0.901931643486023},
{'label': 'shakespearean', 'score': 0.09806833416223526}
]]
Return top label
from transformers import pipeline
classifier = pipeline("text-classification", model="notaphoenix/shakespeare_classifier_model")
classifier("This is a modern sentence!")
[
{'label': 'modern', 'score': 0.901931643486023}
]
- Downloads last month
- 91