readme: ---
  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
  