|
{ |
|
"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 |
|
} |
|
|