Belle II Software development
MaximumEstimatorTrainerSQRT Class Reference
Inheritance diagram for MaximumEstimatorTrainerSQRT:
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 337 of file train.py.

Constructor & Destructor Documentation

◆ __init__()

def __init__ (   self)
Constructor

Reimplemented from FittedGroupedDEDXEstimatorTrainer.

Definition at line 340 of file train.py.

340 def __init__(self):
341 """Constructor"""
342 FittedGroupedDEDXEstimatorTrainer.__init__(self, fit_functions.inverse_sqrt, use_sigma_for_result_fitting=False)
343
344 def train_function(fit_data):
345 """Train on the curated-data highest values whose truth value is known"""
346 max_value = self.use_only_the_highest_values(fit_data, 1).p_bin_centers.values[0]
347
348 return [None, [None, max_value, None]]
349
350
351 self.train_function = train_function
352
353

Member Data Documentation

◆ train_function

train_function

this class's training function

Definition at line 351 of file train.py.


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