predict_results_groups.json

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