CoolFace
Modelpublic

qfrodicio/roberta-finetuned-intention-prediction-es

sourceHugging Faceupdated 3y agoView on Hugging Face
0likes17downloads
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. -->

roberta-finetuned-intention-prediction-es

This model is a fine-tuned version of MMG/mlm-spanish-roberta-base on an unknown dataset. It achieves the following results on the evaluation set:

  • —Loss: 1.9097
  • —Accuracy: 0.6918
  • —Precision: 0.6953
  • —Recall: 0.6918
  • —F1: 0.6848

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
  • —trainbatchsize: 16
  • —evalbatchsize: 16
  • —seed: 42
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —num_epochs: 20

Training results

Training LossEpochStepValidation LossAccuracyPrecisionRecallF1
2.29851.01021.74350.49700.43780.49700.4215
1.33992.02041.42050.58280.58720.58280.5624
0.88933.03061.26990.63930.62760.63930.6192
0.56914.04081.33270.65150.66040.65150.6417
0.38375.05101.38360.65920.67100.65920.6528
0.25436.06121.42530.66410.67030.66410.6528
0.16697.07141.53170.66500.67950.66500.6546
0.11398.08161.59390.67250.67540.67250.6615
0.08059.09181.69870.65940.66960.65940.6518
0.057810.010201.69600.67930.67820.67930.6690
0.037411.011221.75900.68240.68770.68240.6729
0.0312.012241.74250.68420.68590.68420.6785
0.018313.013261.81650.68300.68460.68300.6774
0.015214.014281.83480.68660.69270.68660.6799
0.010915.015301.85620.69400.69670.69400.6855
0.009716.016321.87660.68890.69470.68890.6833
0.007317.017341.87450.69200.69480.69200.6851
0.006218.018361.89440.68950.69190.68950.6825
0.005719.019381.91030.69360.69840.69360.6867
0.005220.020401.90970.69180.69530.69180.6848

Framework versions

  • —Transformers 4.35.2
  • —Pytorch 2.1.0+cu121
  • —Datasets 2.16.1
  • —Tokenizers 0.15.0