roberta-depression-severity-lora / training_summary.json
Akashpaul123's picture
Upload training_summary.json with huggingface_hub
fc79f4b verified
raw
history blame contribute delete
498 Bytes
{
"model_info": {
"model_name": "roberta-base",
"use_lora": true,
"total_parameters": 124648708,
"trainable_parameters": 1183492,
"trainable_percentage": "0.95%"
},
"training_config": {
"epochs": 5,
"batch_size": 16,
"effective_batch_size": 32,
"learning_rate": 0.0003,
"lora_rank": 16
},
"test_results": {
"accuracy": 0.7948897444872244,
"f1": 0.7927007884028759,
"precision": 0.7915559690768426,
"recall": 0.7948897444872244
}
}