Belle II Software  release-08-01-10
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 54 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 60 of file B2A714-DeepContinuumSuppression_MVAModel.py.

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

Member Function Documentation

◆ next_batch()

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

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


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