|
--- |
|
license: apache-2.0 |
|
tags: |
|
- generated_from_trainer |
|
metrics: |
|
- accuracy |
|
- recall |
|
- precision |
|
- f1 |
|
model-index: |
|
- name: distilbert-base-uncased_fine_tuned_title_and_text |
|
results: [] |
|
--- |
|
|
|
<!-- This model card has been generated automatically according to the information the Trainer had access to. You |
|
should probably proofread and complete it, then remove this comment. --> |
|
|
|
# distilbert-base-uncased_fine_tuned |
|
|
|
This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an reddit dataset -for NSFW classification. |
|
It was trained on titles + body_text of submissions. |
|
It achieves the following results on the evaluation set: |
|
- Loss: 1.0159 |
|
- Accuracy: {'accuracy': 0.9095537914043252} |
|
- Recall: {'recall': 0.8936873290793071} |
|
- Precision: {'precision': 0.916024293389395} |
|
- F1: {'f1': 0.9047179605490829} |
|
|
|
## Model description |
|
|
|
More information needed |
|
|
|
## Intended uses & limitations |
|
|
|
More information needed |
|
|
|
## Training and evaluation data |
|
|
|
More information needed |
|
|
|
## Training procedure |
|
|
|
### Training hyperparameters |
|
|
|
The following hyperparameters were used during training: |
|
- learning_rate: 5e-05 |
|
- train_batch_size: 32 |
|
- eval_batch_size: 32 |
|
- seed: 42 |
|
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08 |
|
- lr_scheduler_type: linear |
|
- lr_scheduler_warmup_steps: 1000 |
|
- num_epochs: 15 |
|
|
|
### Training results |
|
|
|
| Training Loss | Epoch | Step | Validation Loss | Accuracy | Recall | Precision | F1 | |
|
|:-------------:|:-----:|:-----:|:---------------:|:--------------------------------:|:------------------------------:|:---------------------------------:|:--------------------------:| |
|
| 0.256 | 1.0 | 2284 | 0.2569 | {'accuracy': 0.9085683000273748} | {'recall': 0.8976754785779398} | {'precision': 0.9107514450867052} | {'f1': 0.9041661884540342} | |
|
| 0.1948 | 2.0 | 4568 | 0.2471 | {'accuracy': 0.9138242540377771} | {'recall': 0.8644029170464904} | {'precision': 0.9518193224592221} | {'f1': 0.9060074047533739} | |
|
| 0.1318 | 3.0 | 6852 | 0.3057 | {'accuracy': 0.914207500684369} | {'recall': 0.8977894257064722} | {'precision': 0.9216282606152767} | {'f1': 0.9095526695526697} | |
|
| 0.0865 | 4.0 | 9136 | 0.4174 | {'accuracy': 0.9047358335614564} | {'recall': 0.8697584320875114} | {'precision': 0.9274605103280681} | {'f1': 0.8976831706456546} | |
|
| 0.0545 | 5.0 | 11420 | 0.4635 | {'accuracy': 0.9095537914043252} | {'recall': 0.8849134001823155} | {'precision': 0.9236441484300666} | {'f1': 0.9038640595903165} | |
|
| 0.0359 | 6.0 | 13704 | 0.5654 | {'accuracy': 0.9071448124828908} | {'recall': 0.8919781221513218} | {'precision': 0.9127798507462687} | {'f1': 0.9022591055786076} | |
|
| 0.0262 | 7.0 | 15988 | 0.5568 | {'accuracy': 0.8994251300301123} | {'recall': 0.900865998176846} | {'precision': 0.8910176941282543} | {'f1': 0.8959147827072356} | |
|
| 0.0181 | 8.0 | 18272 | 0.6846 | {'accuracy': 0.9042430878729811} | {'recall': 0.9026891522333638} | {'precision': 0.898491550413973} | {'f1': 0.9005854601261866} | |
|
| 0.0121 | 9.0 | 20556 | 0.7516 | {'accuracy': 0.9071448124828908} | {'recall': 0.8990428441203282} | {'precision': 0.906896551724138} | {'f1': 0.9029526207370108} | |
|
| 0.0119 | 10.0 | 22840 | 0.8614 | {'accuracy': 0.9050095811661648} | {'recall': 0.9002962625341842} | {'precision': 0.9018376897614427} | {'f1': 0.9010663169299197} | |
|
| 0.0105 | 11.0 | 25124 | 0.7298 | {'accuracy': 0.9105940323022174} | {'recall': 0.8907247037374658} | {'precision': 0.9206218348839948} | {'f1': 0.9054265361672554} | |
|
| 0.0049 | 12.0 | 27408 | 0.9237 | {'accuracy': 0.9101560361346839} | {'recall': 0.8828623518687329} | {'precision': 0.9266834110752302} | {'f1': 0.9042422827799498} | |
|
| 0.0026 | 13.0 | 29692 | 0.9489 | {'accuracy': 0.9066520667944156} | {'recall': 0.8988149498632635} | {'precision': 0.9061458931648478} | {'f1': 0.9024655340083519} | |
|
| 0.0016 | 14.0 | 31976 | 1.0045 | {'accuracy': 0.9099917875718587} | {'recall': 0.8963081130355515} | {'precision': 0.9146511627906977} | {'f1': 0.9053867403314917} | |
|
| 0.0022 | 15.0 | 34260 | 1.0159 | {'accuracy': 0.9095537914043252} | {'recall': 0.8936873290793071} | {'precision': 0.916024293389395} | {'f1': 0.9047179605490829} | |
|
|
|
|
|
### Framework versions |
|
|
|
- Transformers 4.21.0 |
|
- Pytorch 1.12.0+cu113 |
|
- Datasets 2.4.0 |
|
- Tokenizers 0.12.1 |
|
|
