Hugging Face's logo

BramVanroy
/
bert-base-dutch-cased-hebban-reviews5

Text Classification
Transformers
PyTorch
TensorBoard
Dutch
bert
sentiment-analysis
dutch
text
Eval Results
Inference Endpoints
Model card Files Metrics Community
1
Edit model card

bert-base-dutch-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.736704788874575
  • best_model_checkpoint: trained/hebban-reviews5/bert-base-dutch-cased/checkpoint-2000

Test results of best checkpoint

  • accuracy: 0.6071005917159763
  • f1: 0.6050857981600024
  • precision: 0.6167698094913165
  • qwk: 0.7455315835020534
  • recall: 0.6071005917159763

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: 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
Downloads last month
16
Inference Examples
Text Classification
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Evaluation results

View on Papers With Code