Source code for glmdenoise.utils.R2_nom_denom

import numpy as np


def R2_nom_denom(y, yhat):
    """ Calculates the nominator and denomitor for calculating R-squared

    Args:
        y (array): data
        yhat (array): predicted data data

    Returns:
        nominator (float or array), denominator (float or array)
    """
    with np.errstate(divide='ignore', invalid='ignore'):
        nom = np.sum((y - yhat)**2, axis=0)
        denom = np.sum(y**2, axis=0)  # Kendricks denominator
    return nom, denom