Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
TMVA::Experimental::SOFIE::PyKeras Namespace Reference

Functions

RModel Parse (std::string filename, int batch_size=-1)
 Parser function for translating Keras .h5 model into a RModel object.
 

Function Documentation

◆ Parse()

RModel TMVA::Experimental::SOFIE::PyKeras::Parse ( std::string filename,
int batch_size = -1 )

Parser function for translating Keras .h5 model into a RModel object.

Accepts the file location of a Keras model and returns the equivalent RModel object. One can specify as option a batch size that can be used when the input Keras model has not a defined input batch size : e.g. for input = (input_dim,)

Definition at line 12 of file RModelParser_Keras.cxx.