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"
