PlusML
Loading...
Searching...
No Matches
loss_gradient.h
1#ifndef LOSS_GRADIENT_H
2#define LOSS_GRADIENT_H
3
4#include "PlusML/util.h"
5#include <Eigen/Dense>
6
7namespace plusml {
11class EXPORT LossGradient {
12public:
20 virtual Eigen::MatrixXf Compute(const Eigen::MatrixXf& parameters,
21 const Eigen::MatrixXf& X,
22 const Eigen::MatrixXf& y) const = 0;
23
24 virtual ~LossGradient() = default;
25};
26} //namespace plusml
27
28#endif //LOSS_GRADIENT_H
Base class for loss gradient implementations.
Definition loss_gradient.h:11
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.