|
--- |
|
datasets: |
|
- BramVanroy/hebban-reviews |
|
language: |
|
- nl |
|
license: mit |
|
metrics: |
|
- accuracy |
|
- f1 |
|
- precision |
|
- qwk |
|
- recall |
|
model-index: |
|
- name: robbert-v2-dutch-base-hebban-reviews5 |
|
results: |
|
- dataset: |
|
config: filtered_rating |
|
name: BramVanroy/hebban-reviews - filtered_rating - 2.0.0 |
|
revision: 2.0.0 |
|
split: test |
|
type: BramVanroy/hebban-reviews |
|
metrics: |
|
- name: Test accuracy |
|
type: accuracy |
|
value: 0.624457593688363 |
|
- name: Test f1 |
|
type: f1 |
|
value: 0.625518585787774 |
|
- name: Test precision |
|
type: precision |
|
value: 0.6295608657909847 |
|
- name: Test qwk |
|
type: qwk |
|
value: 0.7517620387343015 |
|
- name: Test recall |
|
type: recall |
|
value: 0.624457593688363 |
|
task: |
|
name: sentiment analysis |
|
type: text-classification |
|
tags: |
|
- sentiment-analysis |
|
- dutch |
|
- text |
|
widget: |
|
- text: Wauw, wat een leuk boek! Ik heb me er er goed mee vermaakt. |
|
- text: Nee, deze vond ik niet goed. De auteur doet zijn best om je als lezer mee |
|
te trekken in het verhaal maar mij overtuigt het alleszins niet. |
|
- text: Ik vind het niet slecht maar de schrijfstijl trekt me ook niet echt aan. Het |
|
wordt een beetje saai vanaf het vijfde hoofdstuk |
|
--- |
|
|
|
# robbert-v2-dutch-base-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.7480754124116261 |
|
- best_model_checkpoint: trained/hebban-reviews5/robbert-v2-dutch-base/checkpoint-3000 |
|
|
|
# Test results of best checkpoint |
|
- accuracy: 0.624457593688363 |
|
- f1: 0.625518585787774 |
|
- precision: 0.6295608657909847 |
|
- qwk: 0.7517620387343015 |
|
- recall: 0.624457593688363 |
|
|
|
## Confusion matrix |
|
|
|
![cfm](fig/test_confusion_matrix.png) |
|
|
|
## Normalized confusion matrix |
|
|
|
![norm cfm](fig/test_confusion_matrix_norm.png) |
|
|
|
# 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 |
|
|
|
|