CoolFace
Datasetpublic

dusersad12/SST2-Sentiment-Analysis-Results

SST2 Sentiment Analysis Results Overview This dataset card presents the results of our sentiment analysis experiments on the SST-2 (Stanford Sentiment Treebank) benchmark. We fine-tuned BERT-base with various hyperparameter configurations to find the optimal setup for binary sentiment classification. Experimental Setup All experiments used BERT-base-uncased as the backbone model. We explored different learning rates, batch sizes… See the full description on the dataset page: https://huggingface.co/datasets/dusersad12/SST2-Sentiment-Analysis-Results.

sourceHugging Faceapache-2.0updated 1mo agoView on Hugging Face
0likes56downloads
Dataset Card

SST2 Sentiment Analysis Results

<!-- markdownlint-disable first-line-h1 --> <!-- markdownlint-disable html -->

<div align="center"> <img src="figures/training_curve.png" width="70%" alt="Training Curve" /> </div> <hr>

Overview

This dataset card presents the results of our sentiment analysis experiments on the SST-2 (Stanford Sentiment Treebank) benchmark. We fine-tuned BERT-base with various hyperparameter configurations to find the optimal setup for binary sentiment classification.

<div align="center"> <img width="75%" src="figures/confusion_matrix.png"> </div>

Experimental Setup

All experiments used BERT-base-uncased as the backbone model. We explored different learning rates, batch sizes, dropout rates, and weight decay values. Each configuration was trained on the SST-2 training set and evaluated on the validation set.

Best Run Results

The best performing configuration achieved the following metrics on the SST-2 validation set:

MetricValue
eval_accuracy0.924
eval_f10.921
eval_precision0.933
eval_recall0.910
eval_loss0.234
train_loss0.145

Best Run Configuration

HyperparameterValue
learning_rate5e-05
batch_size32
epochs5
dropout0.10
weight_decay0.01

Comparison Across Runs

<div align="center">

RunLearning RateBatch SizeEpochsDropoutF1 Score
run_0012e-053230.100.858
run_0025e-051630.100.888
run_0031e-043250.200.869
run_0043e-056430.150.904
run_0055e-053250.100.921
run_0062e-051650.200.895
run_0071e-046430.300.851
run_0083e-051650.150.909

</div>

<div align="center"> <img src="figures/f1_comparison.png" width="70%" alt="F1 Comparison" /> </div>

Citation

If you use these results, please cite:

bibtex
@misc{sst2-sentiment-analysis,
  title={SST-2 Sentiment Analysis Experiment Results},
  author={Research Team},
  year={2025}
}

License

This work is licensed under the Apache-2.0 License.