jpherrerap's picture
End of training
b1433e4
raw
history blame
5 kB
{
"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
}