seldonian.models.objectives.gradient_Mean_Squared_Error

gradient_Mean_Squared_Error(model, theta, X, Y, **kwargs)

Gradient of the mean squared error w.r.t. theta

Parameters:
  • model – SeldonianModel instance

  • theta (numpy ndarray) – The parameter weights

  • X (numpy ndarray) – The features

  • Y (numpy ndarray) – The labels

Returns:

vector gradient d(MSE)/dtheta

Return type:

numpy ndarray