def dict_key_editer(var):
    """
    Helper function for changing MIMIC-II specific measures to align with MIMIC-III.

    MIMIC-II has no validation set, hence test losses produced during training.
    """
    if var == "loss_test":
        return "loss_dev"
    else:
        return var