PlusML
|
Base class for loss gradient implementations. More...
#include <loss_gradient.h>
Public Member Functions | |
virtual Eigen::MatrixXf | Compute (const Eigen::MatrixXf ¶meters, const Eigen::MatrixXf &X, const Eigen::MatrixXf &y) const =0 |
Virtual function describing the gradient computation function interface. | |
Base class for loss gradient implementations.
|
pure virtual |
Virtual function describing the gradient computation function interface.
parameters | Parameters of the model |
X | Matrix of samples |
y | Matrix of targets |
Implemented in plusml::MAEGrad, and plusml::MSEGrad.