Inheritance diagram for MVADEDXEstimationTrainer:

Collaboration diagram for MVADEDXEstimationTrainer:

Public Member Functions
	__init__ (self)

	train (self, data)

	test (self, data)

Public Attributes
	tree = tree.DecisionTreeRegressor()
	cached copy of the MVA tool

	dedx_estimator_function = None
	by default, the dE/dx-particle-identification trainer has not run yet

str	dedx_column = "dedx"
	the default data column is 'dedx'

Detailed Description

Train a neural network for dE/dx-based particle identification using multivariate data analysis

Definition at line 373 of file train.py.

Constructor & Destructor Documentation

◆ init()

__init__ ( self )

Constructor

Definition at line 376 of file train.py.

    def __init__(self):
        """Constructor"""
 
        ## cached copy of the MVA tool
        self.tree = tree.DecisionTreeRegressor()
        DEDXEstimationTrainer.__init__(self)
 

Member Function Documentation

◆ test()

test	(		self,
			data )

Get the trained neural-network output value for test data

Reimplemented from DEDXEstimationTrainer.

Definition at line 393 of file train.py.

    def test(self, data):
        """Get the trained neural-network output value for test data"""
 
        test_data = data.copy()
        del test_data["p"]
 
        return self.tree.predict(test_data.values)

◆ train()

train	(		self,
			data )

Train the neural network using curated data

Reimplemented from DEDXEstimationTrainer.

Definition at line 383 of file train.py.

    def train(self, data):
        """Train the neural network using curated data"""
 
        train_data = data.copy()
        del train_data["p"]
 
        p_values = data["p"]
 
        self.tree.fit(train_data.values, p_values.values)
 

Member Data Documentation

◆ dedx_column

str dedx_column = "dedx"

inherited

the default data column is 'dedx'

Definition at line 27 of file train.py.

◆ dedx_estimator_function

dedx_estimator_function = None

inherited

by default, the dE/dx-particle-identification trainer has not run yet

Definition at line 25 of file train.py.

◆ tree

tree = tree.DecisionTreeRegressor()

cached copy of the MVA tool

Definition at line 380 of file train.py.

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

tracking/studies/low_momentum_vxd_estimation/train.py

Public Member Functions

Public Attributes