create_databunch#

radionets.evaluation.utils.create_databunch(data_path, fourier, source_list, batch_size)[source]#

Create a dataloader object, which feeds the data batch-wise

Parameters:
data_pathstr

path to the data

fourierbool

true, if data in Fourier space is used

source_listbool

true, if source_list data is used

batch_sizeint

number of images for one batch

Returns:
DataLoader

dataloader object