Bert Base Uncased Imdb

🚀 bert-base-uncased-imdb

This model is a fine - tuned version of bert - base - uncased on the imdb dataset. It can classify text effectively and achieves high accuracy on the evaluation set.

🚀 Quick Start

This model is ready to use for text classification tasks. You can load it via the Hugging Face Transformers library and start making predictions.

✨ Features

• Fine - tuned: Based on the pre - trained bert - base - uncased model, fine - tuned on the imdb dataset.
• High Accuracy: Achieves an accuracy of 0.9126 on the evaluation set.

📦 Installation

No specific installation steps are provided in the original document.

💻 Usage Examples

No code examples are provided in the original document.

📚 Documentation

Model Performance

This model achieves the following results on the evaluation set:

• Loss: 0.4942
• Accuracy: 0.9126

Model Details

Training Procedure

Training Hyperparameters

The following hyperparameters were used during training:

• learning_rate: 5e - 05
• train_batch_size: 8
• eval_batch_size: 16
• seed: 42
• optimizer: Adam with betas=(0.9,0.999) and epsilon = 1e - 08
• lr_scheduler_type: linear
• lr_scheduler_warmup_steps: 1546
• training_steps: 15468

Training Results

Framework Versions

• Transformers 4.10.2
• Pytorch 1.7.1
• Datasets 1.6.1
• Tokenizers 0.10.3

📄 License

This model is licensed under the Apache 2.0 license.