metadata

datasets:
  - BramVanroy/hebban-reviews
language:
  - nl
license: mit
metrics:
  - accuracy
  - f1
  - precision
  - qwk
  - recall
model-index:
  - name: robbert-v2-dutch-base-hebban-reviews
    results:
      - dataset:
          config: filtered_sentiment
          name: BramVanroy/hebban-reviews - filtered_sentiment - 2.0.0
          revision: 2.0.0
          split: test
          type: BramVanroy/hebban-reviews
        metrics:
          - name: Test accuracy
            type: accuracy
            value: 0.8070512820512821
          - name: Test f1
            type: f1
            value: 0.8144966061997005
          - name: Test precision
            type: precision
            value: 0.8275999429062602
          - name: Test qwk
            type: qwk
            value: 0.7336245557372719
          - name: Test recall
            type: recall
            value: 0.8070512820512821
        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-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.7412639349881154
best_model_checkpoint: trained/hebban-reviews/robbert-v2-dutch-base/checkpoint-3500

Test results of best checkpoint

accuracy: 0.8070512820512821
f1: 0.8144966061997005
precision: 0.8275999429062602
qwk: 0.7336245557372719
recall: 0.8070512820512821

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