Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
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 * *
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 * (see tmva/doc/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.
48/// One can specify as option a batch size that can be used when the input Keras model
49/// has not a defined input batch size : e.g. for input = (input_dim,)
50RModel Parse(std::string filename, int batch_size = -1);
51
52}//PyKeras
53}//SOFIE
54}//Experimental
55}//TMVA
56#endif //TMVA_PYMVA_RMODELPARSER_KERAS
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void char Point_t Rectangle_t WindowAttributes_t Float_t Float_t Float_t Int_t Int_t UInt_t UInt_t Rectangle_t Int_t Int_t Window_t TString Int_t GCValues_t GetPrimarySelectionOwner GetDisplay GetScreen GetColormap GetNativeEvent const char const char dpyName wid window const char font_name cursor keysym reg const char only_if_exist regb h Point_t winding char text const char depth char const char Int_t count const char ColorStruct_t color const char filename
RModel Parse(std::string filename, int batch_size=-1)
Parser function for translatng Keras .h5 model into a RModel object.
create variable transformations