example_main.cc File Reference

#include "keras_model.h"
#include <iostream>

Go to the source code of this file.

Functions
int	main ()

Function Documentation

int main

(

)

Definition at line 17 of file example_main.cc.

           {
  cout << "This is simple example with Keras neural network model loading into C++.\n"
           << "Keras model will be used in C++ for prediction only." << endl;

  DataChunk *sample = new DataChunk2D();
  sample->read_from_file("./example/sample_mnist.dat");
  std::cout << sample->get_3d().size() << std::endl;
  KerasModel m("./example/dumped.nnet");
  m.compute_output(sample);

  return 0;
}