kiduki-v2-stage-gru / metrics.json
misakidesu's picture
Upload KiDUKi v2 stage GRU model
331b0ea verified
Raw
History Blame
2.97 kB
{
"train_size": 245,
"val_size": 52,
"test_size": 52,
"class_weight_mode": "none",
"class_weight_beta": 0.999,
"class_weights": {},
"train_stage_counts": {
"transfer_preparation": 25,
"transfer_execution": 31,
"transfer_stabilization": 25,
"toilet_guidance": 14,
"toilet_cleanup": 20,
"transfer_post_check": 21,
"post_change_dressing": 7,
"diaper_preparation": 7,
"meal_assistance": 12,
"meal_positioning": 13,
"meal_hydration_serving": 24,
"meal_monitoring": 15,
"meal_food_serving": 14,
"toilet_execution": 17
},
"history": [
{
"epoch": 1,
"train": {
"loss": 2.151773452758789,
"stage_acc": 0.3329613097012043
},
"val": {
"loss": 1.4575102925300598,
"stage_acc": 0.546875
}
},
{
"epoch": 2,
"train": {
"loss": 1.1500024646520615,
"stage_acc": 0.6750372052192688
},
"val": {
"loss": 1.0121329128742218,
"stage_acc": 0.690625011920929
}
},
{
"epoch": 3,
"train": {
"loss": 0.7890239730477333,
"stage_acc": 0.774925597012043
},
"val": {
"loss": 0.899249866604805,
"stage_acc": 0.6343750059604645
}
},
{
"epoch": 4,
"train": {
"loss": 0.5551702529191971,
"stage_acc": 0.8431919664144516
},
"val": {
"loss": 0.7968561351299286,
"stage_acc": 0.675000011920929
}
},
{
"epoch": 5,
"train": {
"loss": 0.39808862283825874,
"stage_acc": 0.90234375
},
"val": {
"loss": 0.8136268556118011,
"stage_acc": 0.675000011920929
}
},
{
"epoch": 6,
"train": {
"loss": 0.3225006181746721,
"stage_acc": 0.9040178582072258
},
"val": {
"loss": 0.8159087896347046,
"stage_acc": 0.675000011920929
}
},
{
"epoch": 7,
"train": {
"loss": 0.2238064743578434,
"stage_acc": 0.9295014888048172
},
"val": {
"loss": 0.8080141097307205,
"stage_acc": 0.675000011920929
}
},
{
"epoch": 8,
"train": {
"loss": 0.1912368517369032,
"stage_acc": 0.953125
},
"val": {
"loss": 0.8891348540782928,
"stage_acc": 0.690625011920929
}
},
{
"epoch": 9,
"train": {
"loss": 0.13898206874728203,
"stage_acc": 0.9627976194024086
},
"val": {
"loss": 0.8930219113826752,
"stage_acc": 0.731249988079071
}
},
{
"epoch": 10,
"train": {
"loss": 0.1257116561755538,
"stage_acc": 0.97265625
},
"val": {
"loss": 0.9071497693657875,
"stage_acc": 0.699999988079071
}
}
],
"test": {
"loss": 0.805606484413147,
"stage_acc": 0.746874988079071
}
}