Update README.md
Browse files
README.md
CHANGED
|
@@ -70,34 +70,6 @@ The following hyperparameters were used during training:
|
|
| 70 |
- Datasets 2.1.0
|
| 71 |
- Tokenizers 0.15.1
|
| 72 |
|
| 73 |
-
---
|
| 74 |
-
language: "en"
|
| 75 |
-
tags:
|
| 76 |
-
- distilroberta
|
| 77 |
-
- sentiment
|
| 78 |
-
- NSFW
|
| 79 |
-
- inappropriate
|
| 80 |
-
- spam
|
| 81 |
-
- twitter
|
| 82 |
-
- reddit
|
| 83 |
-
|
| 84 |
-
widget:
|
| 85 |
-
- text: "I like you. You remind me of me when I was young and stupid."
|
| 86 |
-
- text: "I see you’ve set aside this special time to humiliate yourself in public."
|
| 87 |
-
- text: "Have a great weekend! See you next week!"
|
| 88 |
-
|
| 89 |
-
---
|
| 90 |
-
|
| 91 |
-
# Fine-tuned DistilRoBERTa-base for NSFW Classification
|
| 92 |
-
|
| 93 |
-
# Model Description
|
| 94 |
-
|
| 95 |
-
DistilBERT is a transformer model that performs sentiment analysis. I fine-tuned the model on Reddit posts with the purpose of classifying not safe for work (NSFW) content, specifically text that is considered inappropriate and unprofessional. The model predicts 2 classes, which are NSFW or safe for work (SFW).
|
| 96 |
-
|
| 97 |
-
The model is a fine-tuned version of [DistilBERT](https://huggingface.co/docs/transformers/model_doc/distilbert).
|
| 98 |
-
|
| 99 |
-
It was fine-tuned on 14317 Reddit posts pulled from the (Reddit API) [https://praw.readthedocs.io/en/stable/].
|
| 100 |
-
|
| 101 |
# How to Use
|
| 102 |
|
| 103 |
```python
|
|
|
|
| 70 |
- Datasets 2.1.0
|
| 71 |
- Tokenizers 0.15.1
|
| 72 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 73 |
# How to Use
|
| 74 |
|
| 75 |
```python
|