Belle II Software  release-05-02-19
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
 Spectatirs/Quantity to be uncorrelated to.
 
 len
 Length of the data.
 
 index_array
 Index array containing indices from 0 to len.
 
 pointer
 Pointer to the current start of the batch.
 

Detailed Description

Class to create batches for training the Adversary Network.
Once the steps_per_epoch argument is available for the fit function in keras, this class will become obsolete.

Definition at line 38 of file adversary_network.py.

Constructor & Destructor Documentation

◆ __init__()

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

Definition at line 44 of file adversary_network.py.

44  def __init__(self, X, Y, Z):
45  """
46  Init the class
47  :param X: Input Features
48  :param Y: Label Data
49  :param Z: Spectators/Quantity to be uncorrelated to
50  """
51 
52  self.X = X
53 
54  self.Y = Y
55 
56  self.Z = Z
57 
58  self.len = len(Y)
59 
60  self.index_array = np.arange(self.len)
61 
62  self.pointer = 0
63 

Member Function Documentation

◆ next_batch()

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

Definition at line 64 of file adversary_network.py.


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