CoolFace
Modelpublic

cartesinus/fedcsis-intent_baseline-xlm_r-es

sourceHugging Facemitupdated 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. -->

fedcsis-intentbaseline-xlmr-es

This model is a fine-tuned version of xlm-roberta-base on the leyzer-fedcsis dataset.

Test set results:

  • —Accuracy: 0.970738

It achieves the following results on the evaluation set:

  • —Loss: 0.1440
  • —Accuracy: 0.9749

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

Training results

Training LossEpochStepValidation LossAccuracyF1
2.62661.09410.94150.80740.8074
0.72282.018820.48920.89990.8999
0.36123.028230.31100.93460.9346
0.22364.037640.24330.95180.9518
0.14645.047050.19630.95940.9594
0.10566.056460.16980.96670.9667
0.07257.065870.15740.96930.9693
0.06028.075280.14760.97290.9729
0.06199.084690.14740.97430.9743
0.05210.094100.14400.97490.9749

Framework versions

  • —Transformers 4.27.0
  • —Pytorch 1.13.1+cu116
  • —Datasets 2.10.1
  • —Tokenizers 0.13.2

Citation

If you use this model, please cite the following:

@inproceedings{kubis2023caiccaic,
	author={Marek Kubis and Paweł Skórzewski and Marcin Sowański and Tomasz Ziętkiewicz},
	pages={1319–1324},
	title={Center for Artificial Intelligence Challenge on Conversational AI Correctness},
	booktitle={Proceedings of the 18th Conference on Computer Science and Intelligence Systems},
	year={2023},
	doi={10.15439/2023B6058},
	url={http://dx.doi.org/10.15439/2023B6058},
	volume={35},
	series={Annals of Computer Science and Information Systems}
}