t5-english-ner / predict_results_groups.json
imvladikon's picture
init
f6b8a39
raw
history blame
2.23 kB
{
"predict_loss": 0.1715492457151413,
"predict_DATE_precision": 1.0,
"predict_DATE_recall": 0.9444444444444444,
"predict_DATE_f1": 0.9714285714285714,
"predict_DATE_number": 18,
"predict_LOCATION_precision": 0.6666666666666666,
"predict_LOCATION_recall": 0.6666666666666666,
"predict_LOCATION_f1": 0.6666666666666666,
"predict_LOCATION_number": 9,
"predict_ORGANIZATION_precision": 0.961038961038961,
"predict_ORGANIZATION_recall": 0.9487179487179487,
"predict_ORGANIZATION_f1": 0.9548387096774195,
"predict_ORGANIZATION_number": 78,
"predict_OTHER_precision": 0.5185185185185185,
"predict_OTHER_recall": 0.5833333333333334,
"predict_OTHER_f1": 0.5490196078431373,
"predict_OTHER_number": 24,
"predict_PERSON_precision": 0.8679245283018868,
"predict_PERSON_recall": 0.8363636363636363,
"predict_PERSON_f1": 0.8518518518518519,
"predict_PERSON_number": 55,
"predict_QUANTITY_precision": 0.8333333333333334,
"predict_QUANTITY_recall": 0.8928571428571429,
"predict_QUANTITY_f1": 0.8620689655172413,
"predict_QUANTITY_number": 28,
"predict_TITLE_precision": 0.14285714285714285,
"predict_TITLE_recall": 0.1,
"predict_TITLE_f1": 0.11764705882352941,
"predict_TITLE_number": 10,
"predict_overall_precision": 0.8318181818181818,
"predict_overall_recall": 0.8243243243243243,
"predict_overall_f1": 0.8280542986425339,
"predict_overall_accuracy": 0.968184311574328,
"predict_runtime": 19.7294,
"predict_samples_per_second": 1.825,
"predict_steps_per_second": 0.912,
"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
}
}