cli_util.hpp

Go to the documentation of this file.

#ifndef MLPACK_BINDINGS_GO_CLI_UTIL_HPP
#define MLPACK_BINDINGS_GO_CLI_UTIL_HPP

#include <mlpack/core/util/cli.hpp>
#include <mlpack/core/data/dataset_mapper.hpp>

namespace mlpack {
namespace util {

template<typename T>
inline void SetParam(const std::string& identifier, T& value)
{
  CLI::GetParam<T>(identifier) = std::move(value);
}

template<typename T>
inline void SetParamPtr(const std::string& identifier,
                        T* value)
{
  CLI::GetParam<T*>(identifier) = value;
}

template<typename T>
T* GetParamPtr(const std::string& paramName)
{
  return CLI::GetParam<T*>(paramName);
}

inline void EnableVerbose()
{
  Log::Info.ignoreInput = false;
}

inline void DisableVerbose()
{
  Log::Info.ignoreInput = true;
}

inline void DisableBacktrace()
{
  Log::Fatal.backtrace = false;
}

inline void ResetTimers()
{
  // Just get a new object---removes all old timers.
  CLI::GetSingleton().timer.Reset();
}

inline void EnableTimers()
{
  Timer::EnableTiming();
}

} // namespace util
} // namespace mlpack

#endif