{ "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 } }