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 

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

namespace mlpack {
namespace data  {

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

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

extern template bool Load(const std::string&,
                               arma::Mat&,
                               const bool,
                               const bool,
                               const arma::file_type);

// size_t and uword should be one of these three typedefs.
extern template bool Load(const std::string&,
                                        arma::Mat&,
                                        const bool,
                                        const bool,
                                        const arma::file_type);

extern template bool Load(const std::string&,
                                         arma::Mat&,
                                         const bool,
                                         const bool,
                                         const arma::file_type);

extern template bool Load(const std::string&,
                                              arma::Mat&,
                                              const bool,
                                              const bool,
                                              const arma::file_type);

extern template bool Load(const std::string&,
                                 arma::Mat&,
                                 const bool,
                                 const bool,
                                 const arma::file_type);

extern template bool Load(const std::string&,
                                  arma::Mat&,
                                  const bool,
                                  const bool,
                                  const arma::file_type);

extern template bool Load(const std::string&,
                               arma::Mat&,
                               const bool,
                               const bool,
                               const arma::file_type);

extern template bool Load(const std::string&,
                                        arma::SpMat&,
                                        const bool,
                                        const bool);

extern template bool Load(const std::string&,
                                         arma::SpMat&,
                                         const bool,
                                         const bool);

extern template bool Load(const std::string&,
                                              arma::SpMat&,
                                              const bool,
                                              const bool);

extern template bool Load(const std::string&,
                                 arma::SpMat&,
                                 const bool,
                                 const bool);

extern template bool Load(const std::string&,
                                  arma::SpMat&,
                                  const bool,
                                  const bool);

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

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

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

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

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

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

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

extern template bool Load(
    const std::string&,
    arma::Mat&,
    DatasetMapper&,
    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);

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

template<typename eT>
bool Load(const std::vector& files,
          arma::Mat& matrix,
          ImageInfo& info,
          const bool fatal = false);

// Implementation found in load_image.cpp.
bool LoadImage(const std::string& filename,
               arma::Mat& matrix,
               ImageInfo& info,
               const bool fatal = false);

} // 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"
// Include implementation of Load() for images.
#include "load_image_impl.hpp"

#endif