lemonsterpie/finetuned-model-movie-review-sentiment-analysis

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

finetuned-model-movie-review-sentiment-analysis

This model achieves the following results on the evaluation set:

• —Loss: 0.2338
• —Accuracy: 0.95
• —F1: [0.946236559139785, 0.9532710280373832]
• —Precision: [0.946236559139785, 0.9532710280373832]
• —Recall: [0.946236559139785, 0.9532710280373832]
• —Auc: 0.9885

Model description

Binary text classification model that is fine tuned and optimized from JamesH/Moviereviewsentimentanalysismodel.

Intended uses & limitations

Movie review sentiment analysis

Training and evaluation data

Trained on train dataset from datasets library IMDB datasets

Training procedure

200 datapoints from preprocessed train dataset were selected for training.

Training hyperparameters

The following hyperparameters were used during training:

• —learning_rate: 1.2090707659718472e-05
• —trainbatchsize: 4
• —evalbatchsize: 4
• —seed: 42
• —gradientaccumulationsteps: 4
• —totaltrainbatch_size: 16
• —optimizer: Use OptimizerNames.ADAMW8BIT with betas=(0.9,0.999) and epsilon=1e-08 and optimizerargs=No additional optimizer arguments
• —lrschedulertype: linear
• —lrschedulerwarmup_ratio: 0.09356655990374085
• —num_epochs: 4
• —mixedprecisiontraining: Native AMP

Training results

Framework versions

• —Transformers 4.53.1
• —Pytorch 2.7.1+cu118
• —Datasets 3.6.0
• —Tokenizers 0.21.2