README.md

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