PlusML
Loading...
Searching...
No Matches
include
PlusML
gradient
loss_gradient.h
1
#ifndef LOSS_GRADIENT_H
2
#define LOSS_GRADIENT_H
3
4
#include "PlusML/util.h"
5
#include <Eigen/Dense>
6
7
namespace
plusml {
11
class
EXPORT
LossGradient
{
12
public
:
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
plusml::LossGradient
Base class for loss gradient implementations.
Definition
loss_gradient.h:11
plusml::LossGradient::Compute
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.
Generated by
1.10.0