README.md

metadata

license: mit
datasets:
  - notaphoenix/shakespeare_dataset
language:
  - en
metrics:
  - f1
pipeline_tag: text-classification

Emotion English DistilRoBERTa-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 DistilRoBERTa-base.

Application 🚀

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