MedianEstimatorTrainer Class Reference

Inheritance diagram for MedianEstimatorTrainer:

Public Member Functions
def	__init__ (self)

Public Attributes
	train_function
	this class's training function

Detailed Description

Train a neural network for dE/dx-based particle identification using only the median values

Definition at line 292 of file train.py.

Constructor & Destructor Documentation

__init__()

def __init__

(

self

)

Constructor

Reimplemented from FittedGroupedDEDXEstimatorTrainer.

Definition at line 295 of file train.py.

    def __init__(self):
        """Constructor"""
        FittedGroupedDEDXEstimatorTrainer.__init__(self, fit_functions.inverse_squared, use_sigma_for_result_fitting=True)
 
        def train_function(fit_data):
            """Train on the curated-data median values whose truth value is known"""
            weighted_p_values = fit_data.apply(lambda data: [data.p_bin_centers] * int(data.number_of_p_values), axis=1).sum()
            median_value = np.median(weighted_p_values)
            iqr = np.percentile(weighted_p_values, 75) - np.percentile(weighted_p_values, 50)
 
            return [iqr, [None, median_value, None]]
 
        
        self.train_function = train_function
 
 

Member Data Documentation

train_function

train_function

this class's training function

Definition at line 308 of file train.py.

---

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

• tracking/studies/low_momentum_vxd_estimation/train.py