Logo ROOT  
Reference Guide
RModelParser_Keras.h
Go to the documentation of this file.
1// @(#)root/tmva/pymva $Id$
2// Author: Sanjiban Sengupta, 2021
3
4/**********************************************************************************
5 * Project: TMVA - a Root-integrated toolkit for multivariate data analysis *
6 * Package: TMVA *
7 * Web : http://tmva.sourceforge.net *
8 * *
9 * Description: *
10 * Functionality for parsing a saved Keras .H5 model into RModel object *
11 * *
12 * Authors (alphabetical): *
13 * Sanjiban Sengupta <sanjiban.sg@gmail.com> *
14 * *
15 * Copyright (c) 2021: *
16 * CERN, Switzerland *
17 * *
18 * *
19 * Redistribution and use in source and binary forms, with or without *
20 * modification, are permitted according to the terms listed in LICENSE *
21 * (http://tmva.sourceforge.net/LICENSE) *
22 **********************************************************************************/
23
24
25#ifndef TMVA_SOFIE_RMODELPARSER_KERAS
26#define TMVA_SOFIE_RMODELPARSER_KERAS
27
28#include "TMVA/RModel.hxx"
29#include "TMVA/SOFIE_common.hxx"
30#include "TMVA/Types.h"
31#include "TMVA/OperatorList.hxx"
32
33#include "TMVA/PyMethodBase.h"
34
35#include "Rtypes.h"
36#include "TString.h"
37
38
39namespace TMVA{
40namespace Experimental{
41namespace SOFIE{
42namespace PyKeras{
43
44
45/// Parser function for translatng Keras .h5 model into a RModel object.
46/// Accepts the file location of a Keras model and returns the
47/// equivalent RModel object.
48RModel Parse(std::string filename);
49
50}//PyKeras
51}//SOFIE
52}//Experimental
53}//TMVA
54#endif //TMVA_PYMVA_RMODELPARSER_KERAS
RModel Parse(std::string filename)
Parser function for translatng Keras .h5 model into a RModel object.
create variable transformations