vr_class_reward.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_ANN_LOSS_FUNCTIONS_VR_CLASS_REWARD_HPP
#define MLPACK_METHODS_ANN_LOSS_FUNCTIONS_VR_CLASS_REWARD_HPP

#include <mlpack/prereqs.hpp>

namespace mlpack {
namespace ann  {

template<typename MatType = arma::mat>
class VRClassRewardType
{
 public:
  VRClassRewardType(const double scale = 1, const bool sizeAverage = true);

  typename MatType::elem_type Forward(const MatType& input,
                                      const MatType& target);

  void Backward(const MatType& input, const MatType& target, MatType& output);

  template <typename LayerType, typename... Args>
  void Add(Args... args) { network.push_back(new LayerType(args...)); }

  void Add(Layer<MatType>* layer)
  {
    network.push_back(layer);
  }

  const std::vector<Layer<MatType>*>& Network() const { return network; }
  std::vector<Layer<MatType>*>& Network() { return network; }

  bool SizeAverage() const { return sizeAverage; }

  double Scale() const { return scale; }

  template<typename Archive>
  void serialize(Archive& /* ar */, const uint32_t /* version */);

 private:
  double scale;

  bool sizeAverage;

  double reward;

  std::vector<Layer<MatType>*> network;
}; // class VRClassRewardType

// Default typedef for typical `arma::mat` usage.
typedef VRClassRewardType<arma::mat> VRClassReward;

} // namespace ann
} // namespace mlpack

// Include implementation.
#include "vr_class_reward_impl.hpp"

#endif