Belle II Software  release-06-01-15
batch_generator Class Reference

Public Member Functions

def __init__ (self, X, Y, Z)
 
def next_batch (self, batch_size)
 

Public Attributes

 X
 Input Features.
 
 Y
 Label Data.
 
 Z
 Spectators.
 
 len
 Number of events.
 
 index_array
 Index array, which will be shuffled.
 
 pointer
 Pointer for index array.
 

Detailed Description

Class to create batches for training the Adversary Network.
See mva/examples/keras/adversary_network.py for details.

Definition at line 56 of file B2A714-DeepContinuumSuppression_MVAModel.py.

Constructor & Destructor Documentation

◆ __init__()

def __init__ (   self,
  X,
  Y,
  Z 
)
Init the class
:param X: Input Features
:param Y: Label Data
:param Z: Spectaters/Qunatity to be uncorrelated to

Definition at line 62 of file B2A714-DeepContinuumSuppression_MVAModel.py.

62  def __init__(self, X, Y, Z):
63  """
64  Init the class
65  :param X: Input Features
66  :param Y: Label Data
67  :param Z: Spectaters/Qunatity to be uncorrelated to
68  """
69 
70  self.X = X
71 
72  self.Y = Y
73 
74  self.Z = Z
75 
76  self.len = len(Y)
77 
78  self.index_array = np.arange(self.len)
79  np.random.shuffle(self.index_array)
80 
81  self.pointer = 0
82 

Member Function Documentation

◆ next_batch()

def next_batch (   self,
  batch_size 
)
Getting next batch of training data

Definition at line 83 of file B2A714-DeepContinuumSuppression_MVAModel.py.


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