whuir / me-cner Goto Github PK

    for predict_label, gt_label in zip(predict, ground_truth):
        print('debug', predict_label, gt_label)
        if predict_label in [1, 3, 5, 7]:  # B-XXX
            precision_denominator += 1
            pred_category = (predict_label + 1) // 2  # 0-NIL, 1-ORG, 2-LOC, 3-PER, 4-GPE, 5-O
            gt_category = (gt_label + 1) // 2
            confusion_matrix[pred_category][gt_category] += 1

        if gt_label in [1, 3, 5, 7]:  # B-XXX
            recall_denominator += 1
            pred_category = (predict_label + 1) // 2  # 0-NIL, 1-ORG, 2-LOC, 3-PER, 4-GPE, 5-O
            gt_category = (gt_label + 1) // 2
            confusion_matrix[pred_category][gt_category] += 1

        # O, NIL or B-XXX, means the previous NER complete
        if gt_label in [0, 1, 3, 5, 7, 9] and \
                predict_label in [0, 1, 3, 5, 7, 9] and (
                match_candidate[0] > 0 or match_candidate[1] > 0):
            match_candidate = (0, 0)
            numerator += 1
        if predict_label == gt_label and predict_label not in [0, 9]:
            match_candidate = (predict_label, gt_label)
        if predict_label != gt_label:
            match_candidate = (0, 0)