mse_grad.h

#ifndef MSE_GRAD_H
#define MSE_GRAD_H
 
#include "PlusML/gradient/loss_gradient.h"
#include "PlusML/util.h"
 
namespace plusml {
class EXPORT MSEGrad : public LossGradient {
public:
  Eigen::MatrixXf Compute(const Eigen::MatrixXf& w,
                          const Eigen::MatrixXf& X,
                          const Eigen::MatrixXf& y) const override;
 
  void L2Regularization(const float c);
private:
  bool l2_regularization_enabled_ = false;
  float l2_regularization_coefficient_ = 0;
};
} //namespace plusml
 
#endif //MSE_GRAD_H