official-ak/imdb_classification_model

1---2library_name: transformers3license: apache-2.04base_model: distilbert/distilbert-base-uncased5tags:6- generated_from_trainer7metrics:8- accuracy9model-index:10- name: imdb_classification_model11  results: []12---13 14<!-- This model card has been generated automatically according to the information the Trainer had access to. You15should probably proofread and complete it, then remove this comment. -->16 17# imdb_classification_model18 19This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.20It achieves the following results on the evaluation set:21- Loss: 0.237322- Accuracy: 0.932623 24## Model description25 26More information needed27 28## Intended uses & limitations29 30More information needed31 32## Training and evaluation data33 34More information needed35 36## Training procedure37 38### Training hyperparameters39 40The following hyperparameters were used during training:41- learning_rate: 2e-0542- train_batch_size: 1643- eval_batch_size: 1644- seed: 4245- optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments46- lr_scheduler_type: linear47- num_epochs: 248 49### Training results50 51| Training Loss | Epoch | Step | Validation Loss | Accuracy |52|:-------------:|:-----:|:----:|:---------------:|:--------:|53| 0.2128        | 1.0   | 1563 | 0.2082          | 0.9211   |54| 0.1361        | 2.0   | 3126 | 0.2373          | 0.9326   |55 56 57### Framework versions58 59- Transformers 5.13.160- Pytorch 2.11.0+cu12861- Datasets 4.0.062- Tokenizers 0.22.263