def compute_loss(self, y_pred, y_true):
    return np.mean((y_pred - y_true) ** 2)