PlusML
Loading...
Searching...
No Matches
Public Member Functions | List of all members
plusml::LossGradient Class Referenceabstract

Base class for loss gradient implementations. More...

#include <loss_gradient.h>

Public Member Functions

virtual Eigen::MatrixXf Compute (const Eigen::MatrixXf &parameters, const Eigen::MatrixXf &X, const Eigen::MatrixXf &y) const =0
 Virtual function describing the gradient computation function interface.
 

Detailed Description

Base class for loss gradient implementations.

Member Function Documentation

◆ Compute()

virtual Eigen::MatrixXf plusml::LossGradient::Compute ( const Eigen::MatrixXf & parameters,
const Eigen::MatrixXf & X,
const Eigen::MatrixXf & y ) const
pure virtual

Virtual function describing the gradient computation function interface.

Parameters
parametersParameters of the model
XMatrix of samples
yMatrix of targets
Returns
Matrix representing calculated gradient

Implemented in plusml::MAEGrad, and plusml::MSEGrad.


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