Downloads · 30 days
14
2% of all-time downloads
bert-base-multilingual-cased-hebban-reviews5
BramVanroy/bert-base-multilingual-cased-hebban-reviews5
bert-base-multilingual-cased-hebban-reviews5 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: filteredrating - datasetrevision: 2.0.0 - labelcolumn: reviewrating0 - textcolumn: reviewtextwithoutquotes
All-time downloads
722
Public
Repo size
1.5 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin712 MB · 91%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- mit
- Model type
- bert
- Access
- Public
- Created
- Jul 29, 2022
- Updated
- Dec 12, 2022
- SHA
- 873abe60
Try a prompt
Datasets
Model card
From the Hugging Face model README
bert-base-multilingual-cased-hebban-reviews5
Dataset
- dataset_name: BramVanroy/hebban-reviews
- dataset_config: filtered_rating
- dataset_revision: 2.0.0
- labelcolumn: review_rating0
- 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.697825193570947
- best_model_checkpoint: trained/hebban-reviews5/bert-base-multilingual-cased/checkpoint-4500
Test results of best checkpoint
- accuracy: 0.5898668639053254
- f1: 0.5899204480029937
- precision: 0.5971431895675179
- qwk: 0.7050840079198698
- recall: 0.5898668639053254
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: 8159b4c1d5e66b36f68dd263299927ffb8670ebd
- 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
- bert
- License
- mit
- Languages
- nl
- Created
- Jul 29, 2022
- Updated
- Dec 12, 2022