def _underscore_to_cap(_str):
    """
    convenience function taking in an underscore-separated string and turns it into capitalized string
    e.g "policy_loss" -> "Policy loss:"
    """
    spl = _str.split("_")
    spl[0] = spl[0].capitalize()
    return " ".join(spl)