utils_evaluate.py

'''
Adapted from https://github.com/lupantech/ScienceQA
'''

import os
import json
import argparse
import warnings
import pandas as pd
from sentence_transformers import SentenceTransformer
from evaluations import caculate_bleu, caculate_rouge, caculate_similariry

warnings.filterwarnings('ignore')

def get_acc_with_contion(res_pd, key, values):
    if isinstance(values, list):
        total_pd = res_pd[res_pd[key].isin(values)]
    else:
        total_pd = res_pd[res_pd[key] == values]
    correct_pd = total_pd[total_pd['true_false'] == True]
    acc = "{:.2f}".format(len(correct_pd) / len(total_pd) * 100)
    return acc


def get_scores(result_data, rationale_data, results_reference, data_file, img):
    # read result file
    results = result_data
    num = len(results)
    

    # read data file
    sqa_data = json.load(open(data_file))

    # construct pandas data
    sqa_pd = pd.DataFrame(sqa_data).T
    res_pd = sqa_pd[sqa_pd['split'] == 'test']  # test set
    
    
    if img:
        res_pd = res_pd[res_pd["image"] == 'image.png']

    # update data
    for index, row in res_pd.iterrows():

        res_pd.loc[index, 'no_context'] = True if (not row['hint'] and not row['image']) else False
        res_pd.loc[index, 'has_text'] = True if row['hint'] else False
        res_pd.loc[index, 'has_image'] = True if row['image'] else False
        res_pd.loc[index, 'has_text_image'] = True if (row['hint'] and row['image']) else False
        res_pd.loc[index, 'has_no_image'] = False if row['image'] else True

        label = row['answer']
        pred = int(results[index])
        res_pd.loc[index, 'pred'] = pred
        res_pd.loc[index, 'true_false'] = (label == pred)

    # accuracy scores
    acc_average = len(res_pd[res_pd['true_false'] == True]) / num * 100


    # rationale quality

    ## BLEU
    bleu1 = caculate_bleu(rationale_data, results_reference, gram=1)
    bleu4 = caculate_bleu(rationale_data, results_reference, gram=4)

    ## Rouge-L
    rouge = caculate_rouge(rationale_data, results_reference)

    ## Similarity
    model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2').cuda()
    similariry = caculate_similariry(rationale_data, results_reference, model)

    if img:
        scores = {
                "answer":{
                    'acc_natural':
                    get_acc_with_contion(res_pd, 'subject', 'natural science'),
                    'acc_social':
                    get_acc_with_contion(res_pd, 'subject', 'social science'),
                    'acc_language':
                    get_acc_with_contion(res_pd, 'subject', 'language science'),
                    'acc_has_text':
                    get_acc_with_contion(res_pd, 'has_text', True),
                    'acc_has_image':
                    get_acc_with_contion(res_pd, 'has_image', True),
                    'acc_grade_1_6':
                    get_acc_with_contion(res_pd, 'grade', ['grade1', 'grade2', 'grade3', 'grade4', 'grade5', 'grade6']),
                    'acc_grade_7_12':
                    get_acc_with_contion(res_pd, 'grade', ['grade7', 'grade8', 'grade9', 'grade10', 'grade11', 'grade12']),
                    'acc_average':
                    "{:.2f}".format(acc_average),
                },
                "rationale":{
                    'bleu1': bleu1 * 100,
                    'bleu4': bleu4 * 100,
                    'rouge': rouge * 100,
                    'similariry': similariry * 100,
                }
        }
    else:
        scores = {
                "answer":{
                    'acc_natural':
                    get_acc_with_contion(res_pd, 'subject', 'natural science'),
                    'acc_social':
                    get_acc_with_contion(res_pd, 'subject', 'social science'),
                    'acc_language':
                    get_acc_with_contion(res_pd, 'subject', 'language science'),
                    'acc_has_text':
                    get_acc_with_contion(res_pd, 'has_text', True),
                    'acc_has_image':
                    get_acc_with_contion(res_pd, 'has_image', True),
                    'acc_has_no_image':
                    get_acc_with_contion(res_pd, 'has_no_image', True),
                    'acc_no_context':
                    get_acc_with_contion(res_pd, 'no_context', True),
                    'acc_grade_1_6':
                    get_acc_with_contion(res_pd, 'grade', ['grade1', 'grade2', 'grade3', 'grade4', 'grade5', 'grade6']),
                    'acc_grade_7_12':
                    get_acc_with_contion(res_pd, 'grade', ['grade7', 'grade8', 'grade9', 'grade10', 'grade11', 'grade12']),
                    'acc_average':
                    "{:.2f}".format(acc_average),
                },
                "rationale":{
                    'bleu1': bleu1 * 100,
                    'bleu4': bleu4 * 100,
                    'rouge': rouge * 100,
                    'similariry': similariry * 100,
                }
        }

    return scores


def print_scores(scores):
    latex_output = ""
    for key, score in scores.items():
        print(f"{key[4:]}: \t{score}")
        latex_output += f"& {score} "
    latex_output += "\\\\"
    print(latex_output)