[tutorials][tmva] Use different name for header and weights in some S…

…OFIE tutorials
lmoneta · Apr 3, 2024 · 3421389 · 3421389
1 parent 3990d95
commit 3421389
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/tutorials/tmva/TMVA_SOFIE_Inference.py b/tutorials/tmva/TMVA_SOFIE_Inference.py
@@ -31,7 +31,7 @@
 # parse the input Keras model into RModel object
 model = ROOT.TMVA.Experimental.SOFIE.PyKeras.Parse(modelFile)
 
-generatedHeaderFile = modelFile.replace(".h5",".hxx")
+generatedHeaderFile = "generatedSofieHiggsModel.hxx" #modelFile.replace(".h5",".hxx")
 print("Generating inference code for the Keras model from ",modelFile,"in the header ", generatedHeaderFile)
 #Generating inference code
 model.Generate()
@@ -44,7 +44,6 @@
 ROOT.gInterpreter.Declare('#include "' + generatedHeaderFile + '"')
 
 
-generatedHeaderFile = modelFile.replace(".h5",".hxx")
 print("Generating inference code for the Keras model from ",modelFile,"in the header ", generatedHeaderFile)
 #Generating inference
 
@@ -67,7 +66,7 @@
 print("size of data", dataset_size)
 
 #instantiate SOFIE session class 
-session = ROOT.TMVA_SOFIE_Higgs_trained_model.Session()
+session = ROOT.TMVA_SOFIE_Higgs_trained_model.Session("generatedSofieHiggsModel.dat")
 
 hs = ROOT.TH1D("hs","Signal result",100,0,1)
 for i in range(0,dataset_size):

diff --git a/tutorials/tmva/TMVA_SOFIE_RDataFrame.py b/tutorials/tmva/TMVA_SOFIE_RDataFrame.py
@@ -27,14 +27,14 @@
 
 # generating inference code
 model.Generate()
-model.OutputGenerated("Higgs_trained_model.hxx")
+model.OutputGenerated("Higgs_generated_Sofie_model.hxx")
 model.PrintGenerated()
 
 # compile using ROOT JIT trained model
 print("compiling SOFIE model and functor....")
-ROOT.gInterpreter.Declare('#include "Higgs_trained_model.hxx"')
+ROOT.gInterpreter.Declare('#include "Higgs_generated_Sofie_model.hxx"')
 modelName = "Higgs_trained_model"
-ROOT.gInterpreter.Declare('auto sofie_functor = TMVA::Experimental::SofieFunctor<7,TMVA_SOFIE_'+modelName+'::Session>(0);')
+ROOT.gInterpreter.Declare('auto sofie_functor = TMVA::Experimental::SofieFunctor<7,TMVA_SOFIE_'+modelName+'::Session>(0,"Higgs_generated_Sofie_model.dat");')
 
 # run inference over input data
 inputFile = "http://root.cern/files/Higgs_data.root"