t5-english-ner / train_results_groups.json
imvladikon's picture
init
f6b8a39
{
"train_loss": 0.14113017916679382,
"train_COMMERCIAL_ITEM_precision": 0.0,
"train_COMMERCIAL_ITEM_recall": 0.0,
"train_COMMERCIAL_ITEM_f1": 0.0,
"train_COMMERCIAL_ITEM_number": 7,
"train_DATE_precision": 0.9620786516853933,
"train_DATE_recall": 0.9771754636233951,
"train_DATE_f1": 0.9695682944090589,
"train_DATE_number": 701,
"train_EVENT_precision": 0.0,
"train_EVENT_recall": 0.0,
"train_EVENT_f1": 0.0,
"train_EVENT_number": 22,
"train_LOCATION_precision": 0.8862461220268872,
"train_LOCATION_recall": 0.8917793964620188,
"train_LOCATION_f1": 0.8890041493775933,
"train_LOCATION_number": 961,
"train_ORGANIZATION_precision": 0.9199203187250996,
"train_ORGANIZATION_recall": 0.9355753646677472,
"train_ORGANIZATION_f1": 0.9276817999196465,
"train_ORGANIZATION_number": 2468,
"train_OTHER_precision": 0.5351598173515981,
"train_OTHER_recall": 0.541089566020314,
"train_OTHER_f1": 0.5381083562901744,
"train_OTHER_number": 1083,
"train_PERSON_precision": 0.8979591836734694,
"train_PERSON_recall": 0.9413388543823327,
"train_PERSON_f1": 0.9191374663072777,
"train_PERSON_number": 1449,
"train_QUANTITY_precision": 0.8873873873873874,
"train_QUANTITY_recall": 0.9231490159325211,
"train_QUANTITY_f1": 0.9049150206706478,
"train_QUANTITY_number": 1067,
"train_TITLE_precision": 0.22058823529411764,
"train_TITLE_recall": 0.17045454545454544,
"train_TITLE_f1": 0.19230769230769232,
"train_TITLE_number": 264,
"train_overall_precision": 0.8415670814340274,
"train_overall_recall": 0.8515332834704562,
"train_overall_f1": 0.8465208501146292,
"train_overall_accuracy": 0.9713395552737148,
"train_runtime": 701.7841,
"train_samples_per_second": 2.018,
"train_steps_per_second": 1.009,
"model_name": "t5-large",
"dataset_name": "imvladikon/ner_news_small",
"split_instructions": "80:10:10",
"device": "cpu",
"label2id": {
"O": 0,
"B-PERSON": 1,
"I-PERSON": 2,
"B-ORGANIZATION": 3,
"I-ORGANIZATION": 4,
"B-LOCATION": 5,
"I-LOCATION": 6,
"B-TITLE": 7,
"I-TITLE": 8,
"B-EVENT": 9,
"I-EVENT": 10,
"B-QUANTITY": 11,
"I-QUANTITY": 12,
"B-DATE": 13,
"I-DATE": 14,
"B-COMMERCIAL_ITEM": 15,
"I-COMMERCIAL_ITEM": 16,
"B-OTHER": 17,
"I-OTHER": 18
}
}