Prediction on model in cpp

pplonski / keras2cpp

This is a bunch of code to port Keras neural network model into pure C++.

MIT License

679 stars 153 forks source link

Prediction on model in cpp #5

Closed forrestdavis closed 7 years ago

forrestdavis commented 7 years ago

How exactly do you return a prediction for the keras model in the cpp code? Thanks for the help!

pplonski commented 7 years ago

The prediction from Keras model in cpp is returned as DataChunk. Please check method compute_output from KerasModel. The last lines of this method return a prediction.