D-Psyntom / README.md
eperezs's picture
Update README.md
e642fd8 verified
|
raw
history blame
501 Bytes
metadata
library_name: transformers
tags: []

Model Card for Model ID

image/png

---------

checkpoint = "AIMH/mental-roberta-large"

learning_rate = 2e-5

batch_size = 4

num_train_epochs = 5

weight_decay = 0.01

metric_name = "accuracy"

optimizer = AdamW

test_size = 0.2

logging_steps = 25

column_item = "labels_array"