{ "best_metric": 0.7252032520325203, "best_model_checkpoint": "models/nominal-groups-recognition-bert-base-spanish-wwm-cased/checkpoint-4016", "epoch": 5.0, "global_step": 5020, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 1.0, "learning_rate": 1.6000000000000003e-05, "loss": 0.4301, "step": 1004 }, { "epoch": 1.0, "eval_Body_Part_f1": 0.6532887402452621, "eval_Body_Part_number": 413, "eval_Body_Part_precision": 0.6053719008264463, "eval_Body_Part_recall": 0.7094430992736077, "eval_Disease_f1": 0.7027027027027026, "eval_Disease_number": 975, "eval_Disease_precision": 0.6987829614604463, "eval_Disease_recall": 0.7066666666666667, "eval_Family_Member_f1": 0.8421052631578948, "eval_Family_Member_number": 30, "eval_Family_Member_precision": 0.8888888888888888, "eval_Family_Member_recall": 0.8, "eval_Medication_f1": 0.7471264367816093, "eval_Medication_number": 93, "eval_Medication_precision": 0.8024691358024691, "eval_Medication_recall": 0.6989247311827957, "eval_Procedure_f1": 0.5321739130434784, "eval_Procedure_number": 311, "eval_Procedure_precision": 0.5795454545454546, "eval_Procedure_recall": 0.4919614147909968, "eval_loss": 0.30176040530204773, "eval_overall_accuracy": 0.9052301381432105, "eval_overall_f1": 0.668122270742358, "eval_overall_precision": 0.6644951140065146, "eval_overall_recall": 0.6717892425905598, "eval_runtime": 229.9802, "eval_samples_per_second": 3.874, "eval_steps_per_second": 0.487, "step": 1004 }, { "epoch": 2.0, "learning_rate": 1.2e-05, "loss": 0.2384, "step": 2008 }, { "epoch": 2.0, "eval_Body_Part_f1": 0.6966019417475728, "eval_Body_Part_number": 413, "eval_Body_Part_precision": 0.6982968369829684, "eval_Body_Part_recall": 0.6949152542372882, "eval_Disease_f1": 0.7278415015641293, "eval_Disease_number": 975, "eval_Disease_precision": 0.7401908801696713, "eval_Disease_recall": 0.7158974358974359, "eval_Family_Member_f1": 0.8000000000000002, "eval_Family_Member_number": 30, "eval_Family_Member_precision": 0.8, "eval_Family_Member_recall": 0.8, "eval_Medication_f1": 0.7243243243243243, "eval_Medication_number": 93, "eval_Medication_precision": 0.7282608695652174, "eval_Medication_recall": 0.7204301075268817, "eval_Procedure_f1": 0.6035313001605136, "eval_Procedure_number": 311, "eval_Procedure_precision": 0.6025641025641025, "eval_Procedure_recall": 0.6045016077170418, "eval_loss": 0.29026830196380615, "eval_overall_accuracy": 0.9147614164203415, "eval_overall_f1": 0.7002770083102493, "eval_overall_precision": 0.7069351230425056, "eval_overall_recall": 0.6937431394072447, "eval_runtime": 223.1974, "eval_samples_per_second": 3.992, "eval_steps_per_second": 0.502, "step": 2008 }, { "epoch": 3.0, "learning_rate": 8.000000000000001e-06, "loss": 0.1625, "step": 3012 }, { "epoch": 3.0, "eval_Body_Part_f1": 0.7096045197740113, "eval_Body_Part_number": 413, "eval_Body_Part_precision": 0.6652542372881356, "eval_Body_Part_recall": 0.7602905569007264, "eval_Disease_f1": 0.7393316195372751, "eval_Disease_number": 975, "eval_Disease_precision": 0.7412371134020619, "eval_Disease_recall": 0.7374358974358974, "eval_Family_Member_f1": 0.8571428571428571, "eval_Family_Member_number": 30, "eval_Family_Member_precision": 0.9230769230769231, "eval_Family_Member_recall": 0.8, "eval_Medication_f1": 0.7840909090909092, "eval_Medication_number": 93, "eval_Medication_precision": 0.8313253012048193, "eval_Medication_recall": 0.7419354838709677, "eval_Procedure_f1": 0.6220238095238095, "eval_Procedure_number": 311, "eval_Procedure_precision": 0.5789473684210527, "eval_Procedure_recall": 0.6720257234726688, "eval_loss": 0.2948259115219116, "eval_overall_accuracy": 0.9188031610062134, "eval_overall_f1": 0.7150508837707552, "eval_overall_precision": 0.6982217573221757, "eval_overall_recall": 0.7327113062568605, "eval_runtime": 223.9641, "eval_samples_per_second": 3.978, "eval_steps_per_second": 0.5, "step": 3012 }, { "epoch": 4.0, "learning_rate": 4.000000000000001e-06, "loss": 0.1142, "step": 4016 }, { "epoch": 4.0, "eval_Body_Part_f1": 0.7175208581644815, "eval_Body_Part_number": 413, "eval_Body_Part_precision": 0.7065727699530516, "eval_Body_Part_recall": 0.7288135593220338, "eval_Disease_f1": 0.748496993987976, "eval_Disease_number": 975, "eval_Disease_precision": 0.7316356513222331, "eval_Disease_recall": 0.7661538461538462, "eval_Family_Member_f1": 0.8333333333333334, "eval_Family_Member_number": 30, "eval_Family_Member_precision": 0.8333333333333334, "eval_Family_Member_recall": 0.8333333333333334, "eval_Medication_f1": 0.7586206896551724, "eval_Medication_number": 93, "eval_Medication_precision": 0.8148148148148148, "eval_Medication_recall": 0.7096774193548387, "eval_Procedure_f1": 0.6409017713365539, "eval_Procedure_number": 311, "eval_Procedure_precision": 0.6419354838709678, "eval_Procedure_recall": 0.639871382636656, "eval_loss": 0.32472705841064453, "eval_overall_accuracy": 0.920069976473427, "eval_overall_f1": 0.7252032520325203, "eval_overall_precision": 0.7162740899357601, "eval_overall_recall": 0.7343578485181119, "eval_runtime": 223.8603, "eval_samples_per_second": 3.98, "eval_steps_per_second": 0.5, "step": 4016 }, { "epoch": 5.0, "learning_rate": 0.0, "loss": 0.0858, "step": 5020 }, { "epoch": 5.0, "eval_Body_Part_f1": 0.7264260768335273, "eval_Body_Part_number": 413, "eval_Body_Part_precision": 0.6995515695067265, "eval_Body_Part_recall": 0.7554479418886199, "eval_Disease_f1": 0.74435318275154, "eval_Disease_number": 975, "eval_Disease_precision": 0.7451181911613566, "eval_Disease_recall": 0.7435897435897436, "eval_Family_Member_f1": 0.8333333333333334, "eval_Family_Member_number": 30, "eval_Family_Member_precision": 0.8333333333333334, "eval_Family_Member_recall": 0.8333333333333334, "eval_Medication_f1": 0.7745664739884393, "eval_Medication_number": 93, "eval_Medication_precision": 0.8375, "eval_Medication_recall": 0.7204301075268817, "eval_Procedure_f1": 0.6224961479198768, "eval_Procedure_number": 311, "eval_Procedure_precision": 0.5976331360946746, "eval_Procedure_recall": 0.6495176848874598, "eval_loss": 0.3583023250102997, "eval_overall_accuracy": 0.9179586173614044, "eval_overall_f1": 0.7216047709406342, "eval_overall_precision": 0.7129084092126406, "eval_overall_recall": 0.7305159165751921, "eval_runtime": 224.3693, "eval_samples_per_second": 3.971, "eval_steps_per_second": 0.499, "step": 5020 }, { "epoch": 5.0, "step": 5020, "total_flos": 1.0485385474176e+16, "train_loss": 0.20621501573053488, "train_runtime": 33608.7359, "train_samples_per_second": 1.194, "train_steps_per_second": 0.149 } ], "max_steps": 5020, "num_train_epochs": 5, "total_flos": 1.0485385474176e+16, "trial_name": null, "trial_params": null }