get_printable_param_value_impl.hpp

Go to the documentation of this file.

#ifndef MLPACK_BINDINGS_CLI_GET_PRINTABLE_PARAM_VALUE_IMPL_HPP
#define MLPACK_BINDINGS_CLI_GET_PRINTABLE_PARAM_VALUE_IMPL_HPP

#include <mlpack/prereqs.hpp>
#include <mlpack/core/util/param_data.hpp>

namespace mlpack {
namespace bindings {
namespace cli {

template<typename T>
std::string GetPrintableParamValue(
    util::ParamData& /* data */,
    const std::string& input,
    const typename std::enable_if<!arma::is_arma_type<T>::value>::type*,
    const typename std::enable_if<!data::HasSerialize<T>::value>::type*,
    const typename std::enable_if<!std::is_same<T,
        std::tuple<data::DatasetInfo, arma::mat>>::value>::type*)
{
  return input;
}

template<typename T>
std::string GetPrintableParamValue(
    util::ParamData& /* data */,
    const std::string& input,
    const typename std::enable_if<arma::is_arma_type<T>::value>::type*)
{
  return input + ".csv";
}

template<typename T>
std::string GetPrintableParamValue(
    util::ParamData& /* data */,
    const std::string& input,
    const typename std::enable_if<!arma::is_arma_type<T>::value>::type*,
    const typename std::enable_if<data::HasSerialize<T>::value>::type*)
{
  return input + ".bin";
}

template<typename T>
std::string GetPrintableParamValue(
    util::ParamData& /* data */,
    const std::string& input,
    const typename std::enable_if<std::is_same<T,
        std::tuple<data::DatasetInfo, arma::mat>>::value>::type*)
{
  return input + ".arff";
}

} // namespace cli
} // namespace bindings
} // namespace mlpack

#endif