TMVA_SOFIE_Keras_HiggsModel Namespace Reference

Functions
	CreateModel (nlayers=4, nunits=64)
	GenerateCode (modelFile="model.keras")
	PrepareData ()
	TrainModel (model, x, y, name)

Variables
	model = CreateModel(3,64)
	modelFile
str	modelHeaderFile = modelName + ".hxx"
	modelName = GenerateCode(modelFile)
	Step 2 : Parse model and generate inference code with SOFIE.
	session = sofie.Session()
	sofie = getattr(ROOT, 'TMVA_SOFIE_' + modelName)
	Step 3 : Compile the generated C++ model code.
	x = np.random.normal(0,1,7).astype(np.float32)
	x_test
	Step 1 : Create and Train model.
	x_train
	Step 1 : Create and Train model.
	y = session.infer(x)
	y_test
	Step 1 : Create and Train model.
	y_train
	Step 1 : Create and Train model.
	ykeras = model(x.reshape(1,7)).numpy()

Function Documentation

CreateModel()

TMVA_SOFIE_Keras_HiggsModel.CreateModel	(		nlayers = 4,
			nunits = 64 )

Definition at line 19 of file TMVA_SOFIE_Keras_HiggsModel.py.

GenerateCode()

TMVA_SOFIE_Keras_HiggsModel.GenerateCode

(

modelFile = "model.keras"

)

Definition at line 70 of file TMVA_SOFIE_Keras_HiggsModel.py.

PrepareData()

TMVA_SOFIE_Keras_HiggsModel.PrepareData

(

)

Definition at line 32 of file TMVA_SOFIE_Keras_HiggsModel.py.

TrainModel()

TMVA_SOFIE_Keras_HiggsModel.TrainModel	(		model,
			x,
			y,
			name )

Definition at line 63 of file TMVA_SOFIE_Keras_HiggsModel.py.

Variable Documentation

model

TMVA_SOFIE_Keras_HiggsModel.model = CreateModel(3,64)

Definition at line 93 of file TMVA_SOFIE_Keras_HiggsModel.py.

modelFile

TMVA_SOFIE_Keras_HiggsModel.modelFile

Definition at line 94 of file TMVA_SOFIE_Keras_HiggsModel.py.

modelHeaderFile

str TMVA_SOFIE_Keras_HiggsModel.modelHeaderFile = modelName + ".hxx"

Definition at line 101 of file TMVA_SOFIE_Keras_HiggsModel.py.

modelName

TMVA_SOFIE_Keras_HiggsModel.modelName = GenerateCode(modelFile)

Step 2 : Parse model and generate inference code with SOFIE.

Definition at line 100 of file TMVA_SOFIE_Keras_HiggsModel.py.

session

TMVA_SOFIE_Keras_HiggsModel.session = sofie.Session()

Definition at line 115 of file TMVA_SOFIE_Keras_HiggsModel.py.

sofie

TMVA_SOFIE_Keras_HiggsModel.sofie = getattr(ROOT, 'TMVA_SOFIE_' + modelName)

Step 3 : Compile the generated C++ model code.

Step 4: Evaluate the model

Definition at line 114 of file TMVA_SOFIE_Keras_HiggsModel.py.

x

TMVA_SOFIE_Keras_HiggsModel.x = np.random.normal(0,1,7).astype(np.float32)

Definition at line 117 of file TMVA_SOFIE_Keras_HiggsModel.py.

x_test

TMVA_SOFIE_Keras_HiggsModel.x_test

Step 1 : Create and Train model.

Definition at line 91 of file TMVA_SOFIE_Keras_HiggsModel.py.

x_train

TMVA_SOFIE_Keras_HiggsModel.x_train

Step 1 : Create and Train model.

Definition at line 91 of file TMVA_SOFIE_Keras_HiggsModel.py.

y

TMVA_SOFIE_Keras_HiggsModel.y = session.infer(x)

Definition at line 118 of file TMVA_SOFIE_Keras_HiggsModel.py.

y_test

TMVA_SOFIE_Keras_HiggsModel.y_test

Step 1 : Create and Train model.

Definition at line 91 of file TMVA_SOFIE_Keras_HiggsModel.py.

y_train

TMVA_SOFIE_Keras_HiggsModel.y_train

Step 1 : Create and Train model.

Definition at line 91 of file TMVA_SOFIE_Keras_HiggsModel.py.

ykeras

TMVA_SOFIE_Keras_HiggsModel.ykeras = model(x.reshape(1,7)).numpy()

Definition at line 119 of file TMVA_SOFIE_Keras_HiggsModel.py.