plusml::MSEGrad Class Reference

Class implementing Mean Square Error gradient. More...

#include <mse_grad.h>

Public Member Functions
Eigen::MatrixXf	Compute (const Eigen::MatrixXf &w, const Eigen::MatrixXf &X, const Eigen::MatrixXf &y) const override
	Calculate the MSE gradient for given inputs.
void	L2Regularization (const float c)
	Set L2 regularization coefficient.

Detailed Description

Class implementing Mean Square Error gradient.

Member Function Documentation

Compute()

Eigen::MatrixXf plusml::MSEGrad::Compute ( const Eigen::MatrixXf & w, const Eigen::MatrixXf & X, const Eigen::MatrixXf & y ) const

overridevirtual

Calculate the MSE gradient for given inputs.

Parameters

w	Parameters of the model
X	Matrix of samples
y	Matrix of targets

Returns

Matrix representing calculated gradient

Implements plusml::LossGradient.

L2Regularization()

void plusml::MSEGrad::L2Regularization

(

const float

c

)

Set L2 regularization coefficient.

Parameters

c	L2 regularization coefficient

---

The documentation for this class was generated from the following files:

• include/PlusML/gradient/mse_grad/mse_grad.h
• src/gradient/mse_grad/mse_grad.cpp