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.

References keras::KerasModel::compute_output(), keras::DataChunk::get_3d(), and keras::DataChunk::read_from_file().

Referenced by proxy::details::createCollectionProxy(), proxy::CollectionProxyBase< SpacePointWithCharge, MainColl, AuxColl... >::empty(), proxy::CollectionProxyBase< SpacePointWithCharge, MainColl, AuxColl... >::has(), proxy::details::makeCollectionProxy(), proxy::CollectionProxyBase< SpacePointWithCharge, MainColl, AuxColl... >::size(), and proxy::ChargedSpacePointsCollectionProxy< MainColl, AuxColl >::spacePoints().

{
  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;
}