DEDXEstimationTrainer Class Reference

Inheritance diagram for DEDXEstimationTrainer:

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

Public Attributes
	dedx_estimator_function
	by default, the dE/dx-particle-identification trainer has not run yet
	dedx_column
	the default data column is 'dedx'

Detailed Description

Train a neural network for dE/dx-based particle identification

Definition at line 19 of file train.py.

Constructor & Destructor Documentation

__init__()

def __init__

(

self

)

Constructor

Reimplemented in MVADEDXEstimationTrainer, MedianEstimatorTrainerSQRT, MaximumEstimatorTrainerSQRT, LandauEstimatorTrainerSQRT, GaussianEstimatorTrainerSQRT, MedianEstimatorTrainer, MaximumEstimatorTrainer, LandauEstimatorTrainer, and GaussianEstimatorTrainer.

Definition at line 22 of file train.py.

    def __init__(self):
        """Constructor"""
        
        self.dedx_estimator_function = None
        
        self.dedx_column = "dedx"
 

Member Function Documentation

test()

def test	(		self,
			data
	)

Get the trained neural-network output value for test data

Reimplemented in MVADEDXEstimationTrainer.

Definition at line 34 of file train.py.

train()

def train	(		self,
			data
	)

Train on the input data

Reimplemented in MVADEDXEstimationTrainer, and FittedGroupedDEDXEstimatorTrainer.

Definition at line 29 of file train.py.

---

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

• tracking/studies/low_momentum_vxd_estimation/train.py