metadata
library_name: transformers
license: apache-2.0
datasets:
- stanfordnlp/imdb
language:
- en
metrics:
- accuracy
- roc_auc
base_model:
- google-bert/bert-base-uncased
pipeline_tag: text-classification
Model Card for Model ID
Training Details
Training Data
https://huggingface.co/datasets/stanfordnlp/imdb
Training Hyperparameters
-learning_rate = 2e-4 -batch_size = 128 -num_epochs = 3 -weight_decay = 0.01
Evaluation
---Training_loss: 0.355100 ---Validation_loss: 0.330199 ---Accuracy: 0.857 ---Roc_auc: 0.936