| library_name: transformers | |
| tags: [] | |
| # Model Card for Model ID | |
| <!-- Provide a quick summary of what the model is/does. --> | |
|  | |
| ## --------- | |
| checkpoint = "mental/mental-bert-base-uncased" | |
| learning_rate = 2e-5 | |
| batch_size = 16 | |
| num_train_epochs = 5 | |
| weight_decay = 0.01 | |
| metric_name = "accuracy" | |
| optimizer = AdamW | |
| test_size = 0.2 | |
| logging_steps = 25 | |
| column_item = "labels_array" |