Loss.h

Go to the documentation of this file.

#ifndef LOSS_H
#define LOSS_H
#pragma once
 
#include "txeo/Tensor.h"
#include "txeo/types.h"
 
#include <functional>
#include <stdexcept>
namespace txeo {
template <typename T>
class Tensor;
 
template <typename T>
class Loss {
  public:
    Loss(const Loss &) = default;
    Loss(Loss &&) = default;
    Loss &operator=(const Loss &) = default;
    Loss &operator=(Loss &&) = default;
    ~Loss() = default;
 
    explicit Loss(txeo::Tensor<T> &&label, txeo::LossFunc func = txeo::LossFunc::MSE);
 
    explicit Loss(const txeo::Tensor<T> &label, txeo::LossFunc func = txeo::LossFunc::MSE)
        : Loss{label.clone(), func} {};
 
    T get_loss(const txeo::Tensor<T> &pred) const;
 
    void set_loss(txeo::LossFunc func);
 
    T mean_squared_error(const txeo::Tensor<T> &pred) const;
 
    T mean_absolute_error(const txeo::Tensor<T> &pred) const;
 
    T mean_squared_logarithmic_error(const txeo::Tensor<T> &pred) const;
 
    T log_cosh_error(const txeo::Tensor<T> &pred) const;
 
    const txeo::Tensor<T> &label() const { return _label; }
 
    T mse(const txeo::Tensor<T> &pred) const {
      return mean_squared_error(pred);
    }; 
 
    T mae(const txeo::Tensor<T> &pred) const {
      return mean_absolute_error(pred);
    }; 
 
    T msle(const txeo::Tensor<T> &pred) const {
      return mean_squared_logarithmic_error(pred);
    }; 
 
    T lche(const txeo::Tensor<T> &pred) const {
      return log_cosh_error(pred);
    }; 
 
  private:
    Loss() = default;
    txeo::Tensor<T> _label{};
 
    void verify_parameter(const txeo::Tensor<T> &pred) const;
    std::function<T(const txeo::Tensor<T> &)> _loss_func;
};
 
class LossError : public std::runtime_error {
  public:
    using std::runtime_error::runtime_error;
};
 
} // namespace txeo
 
#endif