load.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_DATA_LOAD_HPP
#define MLPACK_CORE_DATA_LOAD_HPP

#include <mlpack/prereqs.hpp>
#include <mlpack/core/util/log.hpp>
#include <string>

#include "format.hpp"
#include "dataset_mapper.hpp"

namespace mlpack {
namespace data  {

template<typename eT>
bool Load(const std::string& filename,
          arma::Mat<eT>& matrix,
          const bool fatal = false,
          const bool transpose = true);

extern template bool Load<int>(const std::string&, arma::Mat<int>&, const bool, const bool);
extern template bool Load<size_t>(const std::string&, arma::Mat<size_t>&, const bool, const bool);
extern template bool Load<float>(const std::string&, arma::Mat<float>&, const bool, const bool);
extern template bool Load<double>(const std::string&, arma::Mat<double>&, const bool, const bool);

#ifndef  _WIN32
extern template bool Load<unsigned long long>(const std::string&, arma::Mat<unsigned long long>&, const bool, const bool);
#endif

template<typename eT>
bool Load(const std::string& filename,
          arma::Col<eT>& vec,
          const bool fatal = false);

template<typename eT>
bool Load(const std::string& filename,
          arma::Row<eT>& rowvec,
          const bool fatal = false);

template<typename eT, typename PolicyType>
bool Load(const std::string& filename,
          arma::Mat<eT>& matrix,
          DatasetMapper<PolicyType>& info,
          const bool fatal = false,
          const bool transpose = true);

extern template bool Load<int, IncrementPolicy>(const std::string&, arma::Mat<int>&,
                                                DatasetMapper<IncrementPolicy>&,
                                                const bool, const bool);

extern template bool Load<size_t, IncrementPolicy>(const std::string&, arma::Mat<size_t>&,
                                                   DatasetMapper<IncrementPolicy>&,
                                                   const bool, const bool);

extern template bool Load<float, IncrementPolicy>(const std::string&, arma::Mat<float>&,
                                                  DatasetMapper<IncrementPolicy>&,
                                                  const bool, const bool);

extern template bool Load<double, IncrementPolicy>(const std::string&, arma::Mat<double>&,
                                                   DatasetMapper<IncrementPolicy>&,
                                                   const bool, const bool);

extern template bool Load<unsigned long long, IncrementPolicy>(const std::string&, arma::Mat<unsigned long long>&,
                                                               DatasetMapper<IncrementPolicy>&,
                                                               const bool, const bool);

template<typename T>
bool Load(const std::string& filename,
          const std::string& name,
          T& t,
          const bool fatal = false,
          format f = format::autodetect);

} // namespace data
} // namespace mlpack

// Include implementation of model-loading Load() overload.
#include "load_model_impl.hpp"
// Include implementation of Load() for vectors.
#include "load_vec_impl.hpp"

#endif