|
{ |
|
"best_metric": 0.7374663072776281, |
|
"best_model_checkpoint": "models/ner-roberta-es-clinical-trials-ner/checkpoint-1506", |
|
"epoch": 3.0, |
|
"global_step": 1506, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 1.0, |
|
"learning_rate": 1.3333333333333333e-05, |
|
"loss": 0.3951, |
|
"step": 502 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"eval_Body_Part_f1": 0.7081339712918661, |
|
"eval_Body_Part_number": 197, |
|
"eval_Body_Part_precision": 0.669683257918552, |
|
"eval_Body_Part_recall": 0.751269035532995, |
|
"eval_Disease_f1": 0.7342256214149141, |
|
"eval_Disease_number": 521, |
|
"eval_Disease_precision": 0.7314285714285714, |
|
"eval_Disease_recall": 0.7370441458733206, |
|
"eval_Family_Member_f1": 0.9166666666666666, |
|
"eval_Family_Member_number": 13, |
|
"eval_Family_Member_precision": 1.0, |
|
"eval_Family_Member_recall": 0.8461538461538461, |
|
"eval_Medication_f1": 0.6493506493506493, |
|
"eval_Medication_number": 37, |
|
"eval_Medication_precision": 0.625, |
|
"eval_Medication_recall": 0.6756756756756757, |
|
"eval_Procedure_f1": 0.608108108108108, |
|
"eval_Procedure_number": 134, |
|
"eval_Procedure_precision": 0.5555555555555556, |
|
"eval_Procedure_recall": 0.6716417910447762, |
|
"eval_loss": 0.2830584943294525, |
|
"eval_overall_accuracy": 0.9144236572118286, |
|
"eval_overall_f1": 0.7071466953250941, |
|
"eval_overall_precision": 0.6861313868613139, |
|
"eval_overall_recall": 0.729490022172949, |
|
"eval_runtime": 5.8394, |
|
"eval_samples_per_second": 76.378, |
|
"eval_steps_per_second": 4.795, |
|
"step": 502 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"learning_rate": 6.666666666666667e-06, |
|
"loss": 0.2123, |
|
"step": 1004 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"eval_Body_Part_f1": 0.7105882352941175, |
|
"eval_Body_Part_number": 197, |
|
"eval_Body_Part_precision": 0.6622807017543859, |
|
"eval_Body_Part_recall": 0.766497461928934, |
|
"eval_Disease_f1": 0.7360308285163777, |
|
"eval_Disease_number": 521, |
|
"eval_Disease_precision": 0.7388781431334622, |
|
"eval_Disease_recall": 0.7332053742802304, |
|
"eval_Family_Member_f1": 0.8148148148148148, |
|
"eval_Family_Member_number": 13, |
|
"eval_Family_Member_precision": 0.7857142857142857, |
|
"eval_Family_Member_recall": 0.8461538461538461, |
|
"eval_Medication_f1": 0.7012987012987014, |
|
"eval_Medication_number": 37, |
|
"eval_Medication_precision": 0.675, |
|
"eval_Medication_recall": 0.7297297297297297, |
|
"eval_Procedure_f1": 0.6825938566552902, |
|
"eval_Procedure_number": 134, |
|
"eval_Procedure_precision": 0.6289308176100629, |
|
"eval_Procedure_recall": 0.746268656716418, |
|
"eval_loss": 0.29119569063186646, |
|
"eval_overall_accuracy": 0.9132166566083283, |
|
"eval_overall_f1": 0.7215053763440861, |
|
"eval_overall_precision": 0.7004175365344467, |
|
"eval_overall_recall": 0.7439024390243902, |
|
"eval_runtime": 5.6078, |
|
"eval_samples_per_second": 79.532, |
|
"eval_steps_per_second": 4.993, |
|
"step": 1004 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"learning_rate": 0.0, |
|
"loss": 0.1686, |
|
"step": 1506 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"eval_Body_Part_f1": 0.7285714285714285, |
|
"eval_Body_Part_number": 197, |
|
"eval_Body_Part_precision": 0.6860986547085202, |
|
"eval_Body_Part_recall": 0.7766497461928934, |
|
"eval_Disease_f1": 0.7490421455938698, |
|
"eval_Disease_number": 521, |
|
"eval_Disease_precision": 0.7476099426386233, |
|
"eval_Disease_recall": 0.7504798464491362, |
|
"eval_Family_Member_f1": 0.8461538461538461, |
|
"eval_Family_Member_number": 13, |
|
"eval_Family_Member_precision": 0.8461538461538461, |
|
"eval_Family_Member_recall": 0.8461538461538461, |
|
"eval_Medication_f1": 0.8266666666666665, |
|
"eval_Medication_number": 37, |
|
"eval_Medication_precision": 0.8157894736842105, |
|
"eval_Medication_recall": 0.8378378378378378, |
|
"eval_Procedure_f1": 0.6758620689655171, |
|
"eval_Procedure_number": 134, |
|
"eval_Procedure_precision": 0.6282051282051282, |
|
"eval_Procedure_recall": 0.7313432835820896, |
|
"eval_loss": 0.28113484382629395, |
|
"eval_overall_accuracy": 0.9174411587205794, |
|
"eval_overall_f1": 0.7374663072776281, |
|
"eval_overall_precision": 0.7177334732423925, |
|
"eval_overall_recall": 0.7583148558758315, |
|
"eval_runtime": 5.4011, |
|
"eval_samples_per_second": 82.576, |
|
"eval_steps_per_second": 5.184, |
|
"step": 1506 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"step": 1506, |
|
"total_flos": 6291231284505600.0, |
|
"train_loss": 0.25864365819599205, |
|
"train_runtime": 931.1664, |
|
"train_samples_per_second": 25.855, |
|
"train_steps_per_second": 1.617 |
|
} |
|
], |
|
"max_steps": 1506, |
|
"num_train_epochs": 3, |
|
"total_flos": 6291231284505600.0, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|