Edit model card

model_gpt2_medium_d_political_bias_ep_2_sqn_a_p_100_v_1

This model is a fine-tuned version of openai-community/gpt2-medium on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 0.4577
  • Balanced Accuracy: 0.7785
  • Accuracy: 0.7784
  • F1 Micro: 0.7784
  • F1 Macro: 0.7781
  • Precision Micro: 0.7784
  • Precision Macro: 0.7801
  • Recall Micro: 0.7784
  • Recall Macro: 0.7785
  • Auc: 0.8765

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: 16
  • eval_batch_size: 64
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 2

Training results

Training Loss Epoch Step Validation Loss Balanced Accuracy Accuracy F1 Micro F1 Macro Precision Micro Precision Macro Recall Micro Recall Macro Auc
0.6433 1.0 727 0.4741 0.7665 0.7663 0.7663 0.7662 0.7663 0.7671 0.7663 0.7665 0.8560
0.4554 2.0 1454 0.4577 0.7785 0.7784 0.7784 0.7781 0.7784 0.7801 0.7784 0.7785 0.8765

Framework versions

  • Transformers 4.44.0
  • Pytorch 2.3.1+cu121
  • Datasets 2.20.0
  • Tokenizers 0.19.1
Downloads last month
0
Safetensors
Model size
355M params
Tensor type
F32
·
Inference API
Unable to determine this model's library. Check the docs .