parameter_type.hpp

Go to the documentation of this file.

 
#ifndef MLPACK_BINDINGS_CLI_PARAMETER_TYPE_HPP
#define MLPACK_BINDINGS_CLI_PARAMETER_TYPE_HPP
 
#include <mlpack/prereqs.hpp>
 
namespace mlpack {
namespace bindings {
namespace cli {
 
// Default: HasSerialize = false.
template<bool HasSerialize, typename T>
struct ParameterTypeDeducer
{
  typedef T type;
};
 
// If we have a serialize() function, then the type is a string.
template<typename T>
struct ParameterTypeDeducer<true, T>
{
  typedef std::string type;
};
 
template<typename T>
struct ParameterType
{
  typedef typename ParameterTypeDeducer<data::HasSerialize<T>::value, T>::type
      type;
};
 
template<typename eT>
struct ParameterType<arma::Col<eT>>
{
  typedef std::string type;
};
 
template<typename eT>
struct ParameterType<arma::Row<eT>>
{
  typedef std::string type;
};
 
template<typename eT>
struct ParameterType<arma::Mat<eT>>
{
  typedef std::string type;
};
 
template<typename eT, typename PolicyType>
struct ParameterType<std::tuple<mlpack::data::DatasetMapper<PolicyType,
                         std::string>, arma::Mat<eT>>>
{
  typedef std::string type;
};
 
} // namespace cli
} // namespace bindings
} // namespace mlpack
 
#endif