UNIQORN -- The Universal Neural-network Interface for Quantum Observable Readout from N-body wavefunctions
DataGenerator.DataGenerator Class Reference
Inheritance diagram for DataGenerator.DataGenerator:
Collaboration diagram for DataGenerator.DataGenerator:

Public Member Functions

def __init__ (self, list_IDs, shuffle=True, batch_size=5)
 
def __len__ (self)
 
def __getitem__ (self, index)
 
def on_epoch_end (self)
 

Public Attributes

 batch_size
 
 list_IDs
 
 shuffle
 
 indexes
 

Private Member Functions

def __data_generation (self, list_IDs_temp, indexes)
 

Detailed Description

Generates data for Keras 

Constructor & Destructor Documentation

◆ __init__()

def DataGenerator.DataGenerator.__init__ (   self,
  list_IDs,
  shuffle = True,
  batch_size = 5 
)
Initialization

Member Function Documentation

◆ __data_generation()

def DataGenerator.DataGenerator.__data_generation (   self,
  list_IDs_temp,
  indexes 
)
private
Here is the call graph for this function:
Here is the caller graph for this function:

◆ __getitem__()

def DataGenerator.DataGenerator.__getitem__ (   self,
  index 
)
Generate one batch of data 
Here is the call graph for this function:

◆ __len__()

def DataGenerator.DataGenerator.__len__ (   self)
Denotes the number of batches per epoch 

◆ on_epoch_end()

def DataGenerator.DataGenerator.on_epoch_end (   self)
Updates indexes after each epoch

Member Data Documentation

◆ batch_size

DataGenerator.DataGenerator.batch_size

◆ indexes

DataGenerator.DataGenerator.indexes

◆ list_IDs

DataGenerator.DataGenerator.list_IDs

◆ shuffle

DataGenerator.DataGenerator.shuffle

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