Downloads · 30 days
25
3% of all-time downloads
xlm-roberta-base-hebban-reviews
BramVanroy/xlm-roberta-base-hebban-reviews
xlm-roberta-base-hebban-reviews is a text classification model from BramVanroy. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
- datasetname: BramVanroy/hebban-reviews - datasetconfig: filteredsentiment - datasetrevision: 2.0.0 - labelcolumn: reviewsentiment - textcolumn: reviewtextwithoutquotes
All-time downloads
914
Public
Parameters
278M
3.4 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin1.1 GB · 48%
Parameter types
How the weights are stored.
F32278M · 100%
Try a prompt
Datasets
Model card
From the Hugging Face model README
xlm-roberta-base-hebban-reviews
Dataset
- dataset_name: BramVanroy/hebban-reviews
- dataset_config: filtered_sentiment
- dataset_revision: 2.0.0
- labelcolumn: review_sentiment
- textcolumn: review_text_without_quotes
Training
- optim: adamw_hf
- learning_rate: 5e-05
- per_device_train_batch_size: 64
- per_device_eval_batch_size: 64
- gradient_accumulation_steps: 1
- max_steps: 5001
- save_steps: 500
- metric_for_best_model: qwk
Best checkedpoint based on validation
- best_metric: 0.741533273748008
- best_model_checkpoint: trained/hebban-reviews/xlm-roberta-base/checkpoint-2000
Test results of best checkpoint
- accuracy: 0.8094674556213017
- f1: 0.812677483587223
- precision: 0.8173602585519025
- qwk: 0.7369243423166991
- recall: 0.8094674556213017
Confusion matrix
Normalized confusion matrix
Environment
- cuda_capabilities: 8.0; 8.0
- cuda_device_count: 2
- cuda_devices: NVIDIA A100-SXM4-80GB; NVIDIA A100-SXM4-80GB
- finetuner_commit: 66294c815326c93682003119534cb72009f558c2
- platform: Linux-4.18.0-305.49.1.el8_4.x86_64-x86_64-with-glibc2.28
- python_version: 3.9.5
- toch_version: 1.10.0
- transformers_version: 4.21.0
- Task
- Text Classification
- Library
- transformers
- Type
- xlm-roberta
- License
- mit
- Languages
- nl
- Created
- Jul 26, 2022
- Updated
- Sep 12, 2023