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