distilbert-multitask / training_args.json
myemfar's picture
Initial upload: rpchat multi-task DistilBERT classifier
138f88f verified
{
"epochs": 5,
"batch_size": 32,
"learning_rate": 3e-05,
"max_length": 64,
"lambda_manip": 0.3,
"num_da_labels": 21,
"da_label_names": [
"accusation",
"acknowledgment",
"action",
"agree",
"command",
"conditional",
"confession",
"disagree",
"emote",
"farewell",
"flirt",
"greeting",
"hedge",
"hostile",
"intent",
"offer",
"opinion",
"out_of_character",
"question",
"statement",
"yes_no_question"
],
"da_train_size": 9765,
"manip_train_size": 12949,
"training_time_seconds": 64.3231897354126,
"da_accuracy": 0.884,
"da_correct": 960,
"da_total": 1086,
"manip_accuracy": 0.9875,
"manip_correct": 1421,
"manip_total": 1439,
"manip_fp": 8,
"manip_fn": 10
}