CoolFace
Modelpublic

buelfhood/conplag1_codet5_ep30_bs16_lr3e-05_l512_s42_ppy_loss

sourceHugging Faceapache-2.0updated 11mo agoView on Hugging Face
0likes39downloads
Model Card

<!-- 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. -->

conplag1codet5ep30bs16lr3e-05l512s42ppyloss

This model is a fine-tuned version of Salesforce/codet5-small on an unknown dataset. It achieves the following results on the evaluation set:

  • —Loss: 0.4496
  • —Accuracy: 0.7883
  • —Recall: 0.8158
  • —Precision: 0.5849
  • —F1: 0.6813
  • —F Beta Score: 0.7274

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: 3e-05
  • —trainbatchsize: 16
  • —evalbatchsize: 16
  • —seed: 42
  • —optimizer: Use OptimizerNames.ADAMWTORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizerargs=No additional optimizer arguments
  • —lrschedulertype: linear
  • —num_epochs: 30

Training results

Training LossEpochStepValidation LossAccuracyRecallPrecisionF1F Beta Score
0.68151.0400.64070.54010.81580.35630.4960.5841
0.72362.0800.71370.44530.92110.32410.47950.5879
0.55343.01200.47530.74450.76320.52730.62370.6708
0.47594.01600.44960.78830.81580.58490.68130.7274
0.42565.02000.50390.81750.55260.72410.62690.5961
0.44356.02400.47480.75180.84210.53330.65310.7148
0.36377.02800.52220.70070.89470.47890.62390.7061

Framework versions

  • —Transformers 4.51.3
  • —Pytorch 2.8.0+cu128
  • —Datasets 3.1.0
  • —Tokenizers 0.21.4