---
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