jpherrerap's picture
End of training
6ce36d7
{
"best_metric": 0.7590687601515972,
"best_model_checkpoint": "models/ner-roberta-es-clinical-trials-ner/checkpoint-1004",
"epoch": 2.0,
"global_step": 1004,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"learning_rate": 2.5e-05,
"loss": 0.3329,
"step": 502
},
{
"epoch": 1.0,
"eval_Body_Part_f1": 0.7268408551068883,
"eval_Body_Part_number": 197,
"eval_Body_Part_precision": 0.6830357142857143,
"eval_Body_Part_recall": 0.7766497461928934,
"eval_Disease_f1": 0.7687861271676301,
"eval_Disease_number": 521,
"eval_Disease_precision": 0.7717601547388782,
"eval_Disease_recall": 0.7658349328214972,
"eval_Family_Member_f1": 0.9230769230769231,
"eval_Family_Member_number": 13,
"eval_Family_Member_precision": 0.9230769230769231,
"eval_Family_Member_recall": 0.9230769230769231,
"eval_Medication_f1": 0.7792207792207791,
"eval_Medication_number": 37,
"eval_Medication_precision": 0.75,
"eval_Medication_recall": 0.8108108108108109,
"eval_Procedure_f1": 0.6689655172413793,
"eval_Procedure_number": 134,
"eval_Procedure_precision": 0.6217948717948718,
"eval_Procedure_recall": 0.7238805970149254,
"eval_loss": 0.2561482787132263,
"eval_overall_accuracy": 0.9219070609535305,
"eval_overall_f1": 0.7462203023758099,
"eval_overall_precision": 0.7273684210526316,
"eval_overall_recall": 0.7660753880266076,
"eval_runtime": 7.4415,
"eval_samples_per_second": 59.934,
"eval_steps_per_second": 3.763,
"step": 502
},
{
"epoch": 2.0,
"learning_rate": 0.0,
"loss": 0.1699,
"step": 1004
},
{
"epoch": 2.0,
"eval_Body_Part_f1": 0.7612293144208038,
"eval_Body_Part_number": 197,
"eval_Body_Part_precision": 0.7123893805309734,
"eval_Body_Part_recall": 0.817258883248731,
"eval_Disease_f1": 0.7704130643611912,
"eval_Disease_number": 521,
"eval_Disease_precision": 0.7711538461538462,
"eval_Disease_recall": 0.7696737044145874,
"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.8378378378378378,
"eval_Medication_number": 37,
"eval_Medication_precision": 0.8378378378378378,
"eval_Medication_recall": 0.8378378378378378,
"eval_Procedure_f1": 0.685512367491166,
"eval_Procedure_number": 134,
"eval_Procedure_precision": 0.6510067114093959,
"eval_Procedure_recall": 0.7238805970149254,
"eval_loss": 0.2660825252532959,
"eval_overall_accuracy": 0.923838261919131,
"eval_overall_f1": 0.7590687601515972,
"eval_overall_precision": 0.7417989417989418,
"eval_overall_recall": 0.7771618625277162,
"eval_runtime": 6.1412,
"eval_samples_per_second": 72.624,
"eval_steps_per_second": 4.559,
"step": 1004
},
{
"epoch": 2.0,
"step": 1004,
"total_flos": 4194154189670400.0,
"train_loss": 0.2513850238693663,
"train_runtime": 633.278,
"train_samples_per_second": 25.344,
"train_steps_per_second": 1.585
}
],
"max_steps": 1004,
"num_train_epochs": 2,
"total_flos": 4194154189670400.0,
"trial_name": null,
"trial_params": null
}