Function mse_criterion

mla/ensemble/base.py:23–30

(y, splits)

def mse_criterion(y, splits):
    y_mean = np.mean(y)
    return -sum(
        [
            np.sum((split - y_mean) ** 2) * (float(split.shape[0]) / y.shape[0])
            for split in splits
        ]
    )


def xgb_criterion(y, left, right, loss):