PlusML
Loading...
Searching...
No Matches
mae_grad.h
1#ifndef MAE_GRAD_H
2#define MAE_GRAD_H
3
4#include "PlusML/gradient/loss_gradient.h"
5#include "PlusML/util.h"
6
7namespace plusml {
11class EXPORT MAEGrad : public LossGradient {
12public:
20 Eigen::MatrixXf Compute(const Eigen::MatrixXf& w,
21 const Eigen::MatrixXf& X,
22 const Eigen::MatrixXf& y) const override;
23};
24} //namespace plusml
25
26#endif //MAE_GRAD_H
Base class for loss gradient implementations.
Definition loss_gradient.h:11
Class implementing Mean Absolute Error gradient.
Definition mae_grad.h:11