Junior/Senior Article Classifier
This model is a fine-tuned version of xlm-roberta-base for classifying sports articles as either Professional or Youth/Junior.
It achieves the following results on the evaluation set:
- Accuracy: [0.964]
- F1 Score: [0.964]
- Precision: [0.967]
- Recall: [0.964]
Model description
The model predicts whether a sports article is about professional sports or youth/junior sports based on the text.
- Label 0: Professional
- Label 1: Youth
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: [2e-5]
- train_batch_size: [8]
- eval_batch_size: [8]
- seed: 42
- optimizer: AdamW
- num_epochs: [20]
Framework versions
- Transformers
- Pytorch
- Datasets
- Tokenizers
- Downloads last month
- 16