cv_function.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_CORE_HPT_CV_FUNCTION_HPP
#define MLPACK_CORE_HPT_CV_FUNCTION_HPP
 
#include <mlpack/core.hpp>
 
namespace mlpack {
namespace hpt {
 
template<typename CVType,
         typename MLAlgorithm,
         size_t TotalArgs,
         typename... BoundArgs>
class CVFunction
{
 public:
  CVFunction(CVType& cv,
             data::DatasetMapper<data::IncrementPolicy, double>& datasetInfo,
             const double relativeDelta,
             const double minDelta,
             const BoundArgs&... args);
 
  double Evaluate(const arma::mat& parameters);
 
  void Gradient(const arma::mat& parameters, arma::mat& gradient);
 
  MLAlgorithm& BestModel() { return bestModel; }
 
 private:
  using BoundArgsTupleType = std::tuple<BoundArgs...>;
 
  static const size_t BoundArgsAmount =
      std::tuple_size<BoundArgsTupleType>::value;
 
  template<size_t BoundArgIndex,
           size_t ParamIndex,
           bool BoundArgsIndexInRange = (BoundArgIndex < BoundArgsAmount)>
  struct UseBoundArg;
 
  CVType& cv;
 
  data::DatasetMapper<data::IncrementPolicy, double> datasetInfo;
 
  BoundArgsTupleType boundArgs;
 
  double bestObjective;
 
  MLAlgorithm bestModel;
 
  double relativeDelta;
 
  double minDelta;
 
  template<size_t BoundArgIndex,
           size_t ParamIndex,
           typename... Args,
           typename = typename
               std::enable_if<(BoundArgIndex + ParamIndex < TotalArgs)>::type>
  inline double Evaluate(const arma::mat& parameters, const Args&... args);
 
  template<size_t BoundArgIndex,
           size_t ParamIndex,
           typename... Args,
           typename = typename
               std::enable_if<BoundArgIndex + ParamIndex == TotalArgs>::type,
           typename = void>
  inline double Evaluate(const arma::mat& parameters, const Args&... args);
 
  template<size_t BoundArgIndex,
           size_t ParamIndex,
           typename... Args,
           typename = typename std::enable_if<
               UseBoundArg<BoundArgIndex, ParamIndex>::value>::type>
  inline double PutNextArg(const arma::mat& parameters, const Args&... args);
 
  template<size_t BoundArgIndex,
           size_t ParamIndex,
           typename... Args,
           typename = typename std::enable_if<
               !UseBoundArg<BoundArgIndex, ParamIndex>::value>::type,
           typename = void>
  inline double PutNextArg(const arma::mat& parameters, const Args&... args);
};
 
 
} // namespace hpt
} // namespace mlpack
 
// Include implementation
#include "cv_function_impl.hpp"
 
#endif