Text Classification
Transformers
TensorBoard
Safetensors
modernbert
Generated from Trainer
text-embeddings-inference
Instructions to use matiashoyl/modernbert-match-user-52977 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use matiashoyl/modernbert-match-user-52977 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="matiashoyl/modernbert-match-user-52977")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("matiashoyl/modernbert-match-user-52977") model = AutoModelForSequenceClassification.from_pretrained("matiashoyl/modernbert-match-user-52977", device_map="auto") - Notebooks
- Google Colab
- Kaggle
modernbert-match-user-52977
This model is a fine-tuned version of answerdotai/ModernBERT-base on the None dataset. It achieves the following results on the evaluation set:
- Loss: 3.1451
- Accuracy: 0.5321
- F1: 0.4862
- Precision: 0.4621
- Recall: 0.5321
- F1 Class 0: 0.4
- F1 Class 1: 0.0
- F1 Class 2: 0.2353
- F1 Class 3: 0.0
- F1 Class 4: 0.768
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
- train_batch_size: 4
- eval_batch_size: 4
- seed: 42
- optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: cosine_with_restarts
- lr_scheduler_warmup_steps: 109
- num_epochs: 10
Training results
| Training Loss | Epoch | Step | Accuracy | F1 | F1 Class 0 | F1 Class 1 | F1 Class 2 | F1 Class 3 | F1 Class 4 | Validation Loss | Precision | Recall |
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 1.3032 | 1.0 | 109 | 0.5046 | 0.3756 | 0.0 | 0.0 | 0.0 | 0.1111 | 0.6968 | 1.2096 | 0.3026 | 0.5046 |
| 1.135 | 2.0 | 218 | 0.4954 | 0.4114 | 0.0 | 0.0 | 0.0 | 0.1429 | 0.7591 | 1.3608 | 0.3518 | 0.4954 |
| 0.8566 | 3.0 | 327 | 0.5505 | 0.4808 | 0.4255 | 0.0 | 0.0 | 0.0 | 0.8 | 1.2995 | 0.4319 | 0.5505 |
| 0.8361 | 4.0 | 436 | 0.5046 | 0.4869 | 0.2632 | 0.1875 | 0.1333 | 0.0 | 0.7863 | 1.4707 | 0.4898 | 0.5046 |
| 0.7833 | 5.0 | 545 | 1.6867 | 0.5413 | 0.5272 | 0.5354 | 0.5413 | 0.3333 | 0.0909 | 0.25 | 0.2308 | 0.7966 |
| 0.4747 | 6.0 | 654 | 2.7277 | 0.5505 | 0.4917 | 0.4593 | 0.5505 | 0.3902 | 0.0 | 0.2353 | 0.0 | 0.7812 |
| 0.1311 | 7.0 | 763 | 3.1451 | 0.5321 | 0.4862 | 0.4621 | 0.5321 | 0.4 | 0.0 | 0.2353 | 0.0 | 0.768 |
Framework versions
- Transformers 4.49.0
- Pytorch 2.6.0
- Datasets 2.21.0
- Tokenizers 0.21.0
- Downloads last month
- 4
Model tree for matiashoyl/modernbert-match-user-52977
Base model
answerdotai/ModernBERT-base