Input/Output (pyvisgen.io)

Introduction

pyvisgen.io contains methods for input and output such as the config handling system and data writers.

Submodules

• Configuration (pyvisgen.io.config)
• Data Writers (pyvisgen.io.datawriters)

Reference/API

pyvisgen.io Package

Classes

Config(*, sampling, polarization, bundle, ...)	Main training configuration.
H5Writer(output_path, dataset_type, **kwargs)	HDF5 file writer for pyvisgen datasets.
FITSWriter(output_path, **kwargs)	FITS file writer for pyvisgen visibility datasets.
PTWriter(output_path, dataset_type, ...)	DataWriter class for saving data in PyTorch (.pt) format.
WDSShardWriter(output_path, *, dataset_type, ...)	WebDataset file writer for pyvisgen datasets.