Belle II Software development
MaximumEstimatorTrainer Class Reference
Inheritance diagram for MaximumEstimatorTrainer:
FittedGroupedDEDXEstimatorTrainer GroupedDEDXEstimationTrainer DEDXEstimationTrainer

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 highest values

Definition at line 275 of file train.py.

Constructor & Destructor Documentation

◆ __init__()

def __init__ (   self)
Constructor

Reimplemented from FittedGroupedDEDXEstimatorTrainer.

Definition at line 278 of file train.py.

278 def __init__(self):
279 """Constructor"""
280 FittedGroupedDEDXEstimatorTrainer.__init__(self, fit_functions.inverse_squared, use_sigma_for_result_fitting=False)
281
282 def train_function(fit_data):
283 """Train on the curated-data highest values whose truth value is known"""
284 max_value = self.use_only_the_highest_values(fit_data, 1).p_bin_centers.values[0]
285
286 return [None, [None, max_value, None]]
287
288
289 self.train_function = train_function
290
291

Member Data Documentation

◆ train_function

train_function

this class's training function

Definition at line 289 of file train.py.


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