Hugging Face's logo

ajrayman
/
Immoderation_continuous

Text Classification
Transformers
Safetensors
roberta
Generated from Trainer
Inference Endpoints
Model card Files Community
Immoderation_continuous / README.md
ajrayman's picture
ajrayman
End of training
b4c4063 verified
preview code
|
raw
history blame contribute delete
1.58 kB
metadata
library_name: transformers
license: mit
base_model: roberta-base
tags:
  - generated_from_trainer
model-index:
  - name: Immoderation_continuous
    results: []

Immoderation_continuous

This model is a fine-tuned version of roberta-base on the None dataset. It achieves the following results on the evaluation set:

  • Loss: 0.0435
  • Rmse: 0.2086
  • Mae: 0.1651
  • Corr: 0.2272

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: 2e-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
  • num_epochs: 3

Training results

Training Loss Epoch Step Validation Loss Rmse Mae Corr
No log 1.0 268 0.0424 0.2060 0.1635 0.2115
0.0485 2.0 536 0.0425 0.2063 0.1647 0.2240
0.0485 3.0 804 0.0435 0.2086 0.1651 0.2272

Framework versions

  • Transformers 4.44.1
  • Pytorch 1.11.0
  • Datasets 2.12.0
  • Tokenizers 0.19.1