metadata

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

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