CoolFace
Modelpublic

Adriana213/xlm-roberta-base-finetuned-panx-all

sourceHugging Facemitupdated 2y agoView on Hugging Face
0likes8downloads
Model Card

xlm-roberta-base-finetuned-panx-all

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

  • —Loss: 0.1758
  • —F1 Score: 0.8558

Model description

This model is a fine-tuned version of xlm-roberta-base on a concatenated dataset combining multiple languages, specifically German (de) and French (fr). The model has been trained for token classification tasks and achieves competitive F1-scores across various languages.

Intended uses

Named Entity Recognition (NER) tasks across multiple languages. Token classification tasks that benefit from multilingual training data.

Limitations

Performance may vary on languages not seen during training. The model is fine-tuned on specific datasets and may require further fine-tuning or adjustments for other tasks or domains.

Training and evaluation data

The model was fine-tuned on a combination of German and French datasets, with the training data shuffled and concatenated to form a multilingual corpus. Additionally, the model was evaluated on multiple languages, showing robust performance across different linguistic datasets.

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • —learning_rate: 5e-05
  • —trainbatchsize: 24
  • —evalbatchsize: 24
  • —seed: 42
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —num_epochs: 3

Training results

Training LossEpochStepValidation LossF1 Score
0.2991.08350.20740.8078
0.15872.016700.17050.8461
0.10123.025050.17580.8558

Evaluation results

The model was evaluated on multiple languages, achieving the following F1-scores:

Evaluated ondefriten
Fine-tune on
de0.86580.70210.68770.5830
each0.86580.84110.81800.6870
all0.86850.86540.86690.7678

Framework versions

  • —Transformers 4.41.1
  • —Pytorch 2.3.0+cu121
  • —Datasets 2.19.1
  • —Tokenizers 0.19.1