output_param.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_UTIL_OUTPUT_PARAM_HPP
#define MLPACK_CORE_UTIL_OUTPUT_PARAM_HPP

#include <mlpack/prereqs.hpp>
#include "param_data.hpp"

namespace mlpack {
namespace util {

template<typename T>
void OutputParamImpl(
    const ParamData& data,
    const typename boost::disable_if<arma::is_arma_type<T>>::type* = 0,
    const typename boost::disable_if<IsStdVector<T>>::type* = 0,
    const typename boost::disable_if<data::HasSerialize<T>>::type* = 0,
    const typename boost::disable_if<std::is_same<T,
        std::tuple<data::DatasetInfo, arma::mat>>>::type* = 0);

template<typename T>
void OutputParamImpl(
    const ParamData& data,
    const typename boost::enable_if<IsStdVector<T>>::type* = 0);

template<typename T>
void OutputParamImpl(
    const ParamData& data,
    const typename boost::enable_if<arma::is_arma_type<T>>::type* = 0);

template<typename T>
void OutputParamImpl(
    const ParamData& data,
    const typename boost::enable_if<data::HasSerialize<T>>::type* = 0);

template<typename T>
void OutputParamImpl(
    const ParamData& data,
    const typename boost::enable_if<std::is_same<T,
        std::tuple<data::DatasetInfo, arma::mat>>>::type* = 0);

template<typename T>
void OutputParam(const ParamData& data)
{
  OutputParamImpl<T>(data);
}

} // namespace util
} // namespace mlpack

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

#endif