Edit model card

xlm-roberta-base-finetuned-panx-fr

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

  • Loss: 0.2750
  • F1 Score: 0.8495

Model description

This model is a fine-tuned version of xlm-roberta-base on the French subset of the PAN-X dataset for Named Entity Recognition (NER). The model has been fine-tuned to perform token classification tasks and is evaluated on its performance in identifying named entities in French text.

Intended uses & limitations

Intended uses:

Named Entity Recognition (NER) tasks specifically for French. Token classification tasks involving French text.

Limitations:

The model's performance is optimized for French and may not generalize well to other languages without further fine-tuning. The model's predictions are based on the data it was trained on and may not handle out-of-domain data as effectively.

Training and evaluation data

The model was fine-tuned on the French subset of the PAN-X dataset, which includes labeled examples of named entities in French text. The evaluation data is a separate portion of the same dataset, used to assess the model's performance.

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 5e-05
  • train_batch_size: 24
  • eval_batch_size: 24
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 3

Training results

Training Loss Epoch Step Validation Loss F1 Score
0.5647 1.0 191 0.3242 0.7728
0.2671 2.0 382 0.2672 0.8202
0.1744 3.0 573 0.2750 0.8495

Framework versions

  • Transformers 4.41.1
  • Pytorch 2.3.0+cu121
  • Datasets 2.19.1
  • Tokenizers 0.19.1
Downloads last month
5
Safetensors
Model size
277M params
Tensor type
F32
·

Finetuned from