Test / training_metadata.json
FrAnKu34t23's picture
Upload merged fine-tuned risk prediction model
35cb202 verified
{
"dataset_file": "construction_risk_natural_scenarios.jsonl",
"training_samples": 551,
"cause_weights": {
"Caught In Between": 8.193548387096774,
"Struck By Object": 3.097560975609756,
"Electrocution": 3.4324324324324325,
"Falls": 1.0,
"Transportation Incidents": 3.4324324324324325,
"Other/Unspecified": 12.7,
"Contact With Objects": 16.933333333333334,
"Exposure Harmful Substances": 254.0
},
"optimization_config": {
"learning_rate": 0.0001,
"batch_size": 4,
"max_length": 256,
"epochs": 8,
"lora_r": 8,
"lora_alpha": 16,
"label_smoothing": 0.05,
"gradient_clipping": 1.0
}
}