BramVanroy's picture
Update README.md
c1fa7b2
metadata
datasets:
  - BramVanroy/hebban-reviews
language:
  - nl
license: mit
metrics:
  - accuracy
  - f1
  - precision
  - qwk
  - recall
model-index:
  - name: bert-base-dutch-cased-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.8042406311637081
          - name: Test f1
            type: f1
            value: 0.8125977499178383
          - name: Test precision
            type: precision
            value: 0.8283602308368182
          - name: Test qwk
            type: qwk
            value: 0.7301452890386257
          - name: Test recall
            type: recall
            value: 0.8042406311637081
        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

bert-base-dutch-cased-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.732569302631819
  • best_model_checkpoint: trained/hebban-reviews/bert-base-dutch-cased/checkpoint-3000

Test results of best checkpoint

  • accuracy: 0.8042406311637081
  • f1: 0.8125977499178383
  • precision: 0.8283602308368182
  • qwk: 0.7301452890386257
  • recall: 0.8042406311637081

Confusion matrix

cfm

Normalized confusion matrix

norm cfm

Environment

  • cuda_capabilities: 8.0; 8.0
  • cuda_device_count: 2
  • cuda_devices: NVIDIA A100-SXM4-80GB; NVIDIA A100-SXM4-80GB
  • finetuner_commit: 48bb3434fa8bbfc9b2d0061ca6c8feb87f78a7ef
  • 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