MVADEDXEstimationTrainer Class Reference

Inheritance diagram for MVADEDXEstimationTrainer:

Collaboration diagram for MVADEDXEstimationTrainer:

Public Member Functions
def	__init__ (self)
def	train (self, data)
def	test (self, data)

Public Attributes
	tree
	cached copy of the MVA tool

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__()

def __init__

(

self

)

Constructor

Reimplemented from DEDXEstimationTrainer.

Definition at line 376 of file train.py.

    def __init__(self):
        """Constructor"""
 
        
        self.tree = tree.DecisionTreeRegressor()
        DEDXEstimationTrainer.__init__(self)
 

Member Function Documentation

test()

def 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()

def 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

tree

tree

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