metadata

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.

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}
]