DiyRex's picture
v9 — distilbert emotion classifier (8 labels, 0.9356 acc)
f23310b verified
{
"base_model": "/home/zuselk/Projects/models/distilbert-base-uncased",
"labels": [
"sadness",
"joy",
"love",
"anger",
"fear",
"surprise",
"disgust",
"calm"
],
"test_metrics": {
"eval_loss": 0.2372336983680725,
"eval_accuracy": 0.9356,
"eval_runtime": 0.4577,
"eval_samples_per_second": 5461.578,
"eval_steps_per_second": 87.385,
"epoch": 3.0
}
}