def compute_metrics()

in biolm/utils_classification.py [0:0]

• 58 lines of code
• 28 McCabe index (conditional complexity)

def compute_metrics(task_name, preds, labels, examples):
    assert len(preds) == len(labels) == len(examples)
    if task_name == "medsts":
        return pearson_and_spearman(preds, labels)
    elif task_name == "biosses":
        return pearson_and_spearman(preds, labels)
    elif task_name == "hoc":
        return hoc_get_p_r_f_arrary(preds, labels, examples)
    elif task_name == "mednli":
        return multiclass_acc_and_f1(preds, labels)
    elif task_name =='chemprot':
        return chemprot_eval(preds, labels)
    elif task_name == 'gad1':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'gad2':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'gad3':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'gad4':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'gad5':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'gad6':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'gad7':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'gad8':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'gad9':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'gad10':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'euadr1':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'euadr2':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'euadr3':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'euadr4':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'euadr5':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'euadr6':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'euadr7':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'euadr8':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'euadr9':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'euadr10':
        return acc_p_r_and_f1(preds, labels)
    elif task_name == 'ddi':
        return ddi_eval(preds, labels)
    elif task_name == "i2b22010":
        return i2b22010re_eval(preds, labels)
    else:
        raise KeyError(task_name)