Bert Base Cased Finetuned Stsb

🚀 bert-base-cased-finetuned-stsb

This model is a fine - tuned version of [bert - base - cased](https://huggingface.co/bert - base - cased) on the GLUE STSB dataset. It can effectively compare text similarity, providing high - accuracy evaluation scores on the GLUE STSB dataset, which is of great value for text classification and semantic similarity tasks.

🚀 Quick Start

This model is a fine - tuned version of [bert - base - cased](https://huggingface.co/bert - base - cased) on the GLUE STSB dataset. It achieves the following results on the evaluation set:

• Loss: 0.4861
• Pearson: 0.8926
• Spearmanr: 0.8898
• Combined Score: 0.8912

The model was fine - tuned to compare [google/fnet - base](https://huggingface.co/google/fnet - base) as introduced in this paper against [bert - base - cased](https://huggingface.co/bert - base - cased).

📦 Installation

This model is trained using the [run_glue](https://github.com/huggingface/transformers/blob/master/examples/pytorch/text - classification/run_glue.py) script. The following command was used:

#!/usr/bin/bash

python ../run_glue.py \\n  --model_name_or_path bert-base-cased \\n  --task_name stsb \\n  --do_train \\n  --do_eval \\n  --max_seq_length 512 \\n  --per_device_train_batch_size 16 \\n  --learning_rate 2e-5 \\n  --num_train_epochs 3 \\n  --output_dir bert-base-cased-finetuned-stsb \\n  --push_to_hub \\n  --hub_strategy all_checkpoints \\n  --logging_strategy epoch \\n  --save_strategy epoch \\n  --evaluation_strategy epoch \\n

Training hyperparameters

The following hyperparameters were used during training:

• learning_rate: 2e - 05
• train_batch_size: 16
• eval_batch_size: 8
• seed: 42
• optimizer: Adam with betas=(0.9,0.999) and epsilon = 1e - 08
• lr_scheduler_type: linear
• num_epochs: 3.0

Training results

Framework versions

• Transformers 4.11.0.dev0
• Pytorch 1.9.0
• Datasets 1.12.1
• Tokenizers 0.10.3

📚 Documentation

Model Information

Model Index

• Name: bert-base-cased-finetuned-stsb
• Results:
  • Task:
    • Name: Text Classification
    • Type: text - classification
  • Dataset:
    • Name: GLUE STSB
    • Type: glue
    • Args: stsb
  • Metrics:
    • Name: Spearmanr
    • Type: spearmanr
    • Value: 0.8897907271421561

📄 License

This model is licensed under the Apache - 2.0 license.