CoolFace
Modelpublic

BaxterAI/finetuning-sentiment-model-3000-samples

sourceHugging Faceapache-2.0updated 4y agoView on Hugging Face
1likes113downloads
README.md75 linesDownload Raw Back to root
1---2license: apache-2.03tags:4- generated_from_trainer5datasets:6- amazon_polarity7metrics:8- accuracy9- f110model-index:11- name: finetuning-sentiment-model-3000-samples12  results:13  - task:14      name: Text Classification15      type: text-classification16    dataset:17      name: amazon_polarity18      type: amazon_polarity19      args: amazon_polarity20    metrics:21    - name: Accuracy22      type: accuracy23      value: 0.922524    - name: F125      type: f126      value: 0.924081632653061227---28 29<!-- This model card has been generated automatically according to the information the Trainer had access to. You30should probably proofread and complete it, then remove this comment. -->31 32# finetuning-sentiment-model-3000-samples33 34This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the amazon_polarity dataset.35It achieves the following results on the evaluation set:36- Loss: 0.817037- Accuracy: 0.922538- F1: 0.924139 40## Model description41 42More information needed43 44## Intended uses & limitations45 46More information needed47 48## Training and evaluation data49 50More information needed51 52## Training procedure53 54### Training hyperparameters55 56The following hyperparameters were used during training:57- learning_rate: 3e-0558- train_batch_size: 1659- eval_batch_size: 1660- seed: 4261- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-0862- lr_scheduler_type: linear63- num_epochs: 2064 65### Training results66 67 68 69### Framework versions70 71- Transformers 4.19.272- Pytorch 1.11.0+cu11373- Datasets 2.2.274- Tokenizers 0.12.175