piotrtrochim/action_from_text_model

1---2license: apache-2.03tags:4- generated_from_trainer5datasets:6- imdb7metrics:8- accuracy9model-index:10- name: action_from_text_model11  results:12  - task:13      name: Text Classification14      type: text-classification15    dataset:16      name: imdb17      type: imdb18      config: plain_text19      split: test20      args: plain_text21    metrics:22    - name: Accuracy23      type: accuracy24      value: 0.9307625---26 27<!-- This model card has been generated automatically according to the information the Trainer had access to. You28should probably proofread and complete it, then remove this comment. -->29 30# action_from_text_model31 32This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the imdb dataset.33It achieves the following results on the evaluation set:34- Loss: 0.231035- Accuracy: 0.930836 37## Model description38 39More information needed40 41## Intended uses & limitations42 43More information needed44 45## Training and evaluation data46 47More information needed48 49## Training procedure50 51### Training hyperparameters52 53The following hyperparameters were used during training:54- learning_rate: 2e-0555- train_batch_size: 1656- eval_batch_size: 1657- seed: 4258- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-0859- lr_scheduler_type: linear60- num_epochs: 261 62### Training results63 64| Training Loss | Epoch | Step | Validation Loss | Accuracy |65|:-------------:|:-----:|:----:|:---------------:|:--------:|66| 0.231         | 1.0   | 1563 | 0.1863          | 0.9277   |67| 0.1518        | 2.0   | 3126 | 0.2310          | 0.9308   |68 69 70### Framework versions71 72- Transformers 4.30.273- Pytorch 2.0.1+cu11874- Datasets 2.13.175- Tokenizers 0.13.376