mnist.h

Go to the documentation of this file.

#ifndef MNIST_H
#define MNIST_H
 
#include <vector>
#include <string>
#include <eigen3/Eigen/Dense>
 
class mnist {
private:
  std::vector<std::vector<double>> m_images;
  std::vector<int> m_labels;
  int m_size;
  int m_rows;
  int m_cols;
 
  void load_images(std::string file, int num=0);
  void load_labels(std::string file, int num=0);
  int  to_int(char* p);
  
  
public:
  mnist(std::string image_file, std::string label_file, int num);
  mnist(std::string image_file, std::string label_file);
  ~mnist();
 
  int size() { return m_size; }
  int rows() { return m_rows; }
  int cols() { return m_cols; }
 
  std::vector<double> images(int id) { return m_images[id]; }
  int labels(int id) { return m_labels[id]; }
 
  struct {
    Eigen::MatrixXd images;
    Eigen::MatrixXd labels;
  } data;
 
};
 
#endif