loss.h

Go to the documentation of this file.

 
#pragma once
 
#include <Eigen/Core>
 
namespace slope {
 
class Loss
{
public:
  virtual ~Loss() = default;
 
  virtual double loss(const Eigen::MatrixXd& eta, const Eigen::MatrixXd& y) = 0;
 
  virtual double dual(const Eigen::MatrixXd& theta,
                      const Eigen::MatrixXd& y,
                      const Eigen::VectorXd& w) = 0;
 
  Eigen::MatrixXd residual(const Eigen::MatrixXd& eta,
                           const Eigen::MatrixXd& y);
  // TODO: This function could be removed since it is just
  // g^{-1}(eta) - y for all families
 
  virtual Eigen::MatrixXd hessianDiagonal(const Eigen::MatrixXd& eta) = 0;
 
  virtual void updateWeightsAndWorkingResponse(Eigen::MatrixXd& w,
                                               Eigen::MatrixXd& z,
                                               const Eigen::MatrixXd& eta,
                                               const Eigen::MatrixXd& y);
 
  virtual Eigen::MatrixXd preprocessResponse(const Eigen::MatrixXd& y) = 0;
 
  virtual void updateIntercept(Eigen::VectorXd& beta0,
                               const Eigen::MatrixXd& eta,
                               const Eigen::MatrixXd& y)
  {
    Eigen::MatrixXd residual = this->residual(eta, y);
    beta0 -= residual.colwise().mean() / this->lipschitz_constant;
  };
  // TODO: This function is not currently used anywhere and could
  // probably be removed.
 
  virtual Eigen::MatrixXd link(const Eigen::MatrixXd& mu) = 0;
 
  virtual Eigen::MatrixXd inverseLink(const Eigen::MatrixXd& eta) = 0;
 
  virtual Eigen::MatrixXd predict(const Eigen::MatrixXd& eta) = 0;
 
  virtual double deviance(const Eigen::MatrixXd& eta, const Eigen::MatrixXd& y)
  {
    return 2.0 * (loss(eta, y) - loss(link(y), y));
  }
 
protected:
  explicit Loss(double lipschitz_constant)
    : lipschitz_constant(lipschitz_constant)
  {
  }
 
private:
  const double lipschitz_constant;
};
 
} // namespace slope