TMVAHelper.TMVAHelperXGB Class Reference

Public Member Functions
	__init__ (self, model_input, model_name, variables=[])
	run_inference (self, df, col_name="mva_score")

Public Attributes
list	variables = [str(var.GetString()) for var in variables_]
	model_input = model_input
	model_name = model_name
	nthreads = ROOT.GetThreadPoolSize()
	tmva_helper
str	var_col = f"tmva_vars_{self.model_name}"

Constructor & Destructor Documentation

__init__()

TMVAHelper.TMVAHelperXGB.__init__	(		self,
			model_input,
			model_name,
			variables = [] )

Member Function Documentation

run_inference()

TMVAHelper.TMVAHelperXGB.run_inference	(		self,
			df,
			col_name = "mva_score" )

Member Data Documentation

model_input

TMVAHelper.TMVAHelperXGB.model_input = model_input

model_name

TMVAHelper.TMVAHelperXGB.model_name = model_name

nthreads

TMVAHelper.TMVAHelperXGB.nthreads = ROOT.GetThreadPoolSize()

tmva_helper

TMVAHelper.TMVAHelperXGB.tmva_helper

Initial value:

=  ROOT.tmva_helper_xgb(self.model_input,
                                                self.model_name,
                                                self.nthreads)

var_col

str TMVAHelper.TMVAHelperXGB.var_col = f"tmva_vars_{self.model_name}"

variables

list TMVAHelper.TMVAHelperXGB.variables = [str(var.GetString()) for var in variables_]

---

The documentation for this class was generated from the following file:

• addons/TMVAHelper/python/TMVAHelper.py