`h0rton.trainval_data`¶

Submodules¶

Package Contents¶

Classes¶

`XYData`	Represents the XYData used to train or validate the BNN
`TDLMCData`	Represents the XYData used to train or validate the BNN

class h0rton.trainval_data.XYData(is_train, Y_cols, float_type, define_src_pos_wrt_lens, rescale_pixels, log_pixels, add_pixel_noise, eff_exposure_time, train_Y_mean=None, train_Y_std=None, train_baobab_cfg_path=None, val_baobab_cfg_path=None, for_cosmology=False, rescale_pixels_type='whiten_pixels')[source]¶

Bases: torch.utils.data.Dataset

Represents the XYData used to train or validate the BNN

__getitem__(self, index)¶

__len__(self)¶

class h0rton.trainval_data.TDLMCData(data_cfg, rung_i=2)[source]¶