io_util.hpp

Go to the documentation of this file.

#ifndef MLPACK_BINDINGS_GO_IO_UTIL_HPP
#define MLPACK_BINDINGS_GO_IO_UTIL_HPP

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

namespace mlpack {
namespace util {

template<typename T>
inline void SetParam(util::Params& p,
                     const std::string& identifier,
                     T& value)
{
  p.Get<T>(identifier) = std::move(value);
}

template<typename T>
inline void SetParamPtr(util::Params& p,
                        const std::string& identifier,
                        T* value)
{
  p.Get<T*>(identifier) = value;
}

template<typename T>
T* GetParamPtr(util::Params& p, const std::string& paramName)
{
  return p.Get<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()
{
  Timer::ResetAll();
}

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

} // namespace util
} // namespace mlpack

#endif