Belall87's picture
Upload folder using huggingface_hub
c18b08b verified
{
"model_version": "V2",
"architecture": "VGG16 + GAP + BN + Dense(512,256)",
"input_size": "224x224x3",
"classes": [
"angry",
"normal",
"rested",
"sad",
"surprised"
],
"num_classes": 5,
"training_samples": 11513,
"validation_samples": 2880,
"V1_accuracy": 0.8118,
"V2_accuracy": 0.8219,
"V2_loss": 0.3639,
"improvement": 0.0101,
"per_class_accuracy": {
"angry": 0.9677,
"normal": 0.6082,
"rested": 0.6611,
"sad": 0.8484,
"surprised": 0.9952
},
"training_phases": {
"phase1": {
"epochs": 15,
"lr": 0.0001,
"frozen": "all VGG16"
},
"phase2": {
"epochs": 25,
"lr": 1e-05,
"unfrozen": "last 8 layers"
}
}
}