drw_dataset
#
Module Contents#
Classes#
- class drw_dataset.DRWDataset(params_sampler, out_dir, num_samples, is_training, transform_x_func=lambda x: ..., transform_y_func=lambda x: ..., prestored_bandpasses=list('ugrizy'), seed=123, obs_kwargs={})[source]#
Bases:
torch.utils.data.Dataset
- _generate_light_curve(index, log_rf_tau, log_sf_inf, mean, z)[source]#
Generate a single light curve in a given filter. Rendering is done in the rest frame, with the input params assumed to be in the rest frame.
- indexint
index within the dataset
- log_rf_taufloat
log10 of rest-frame timescale in days
- log_sf_inffloat
log10 of rest-frame asymptotic amplitude in mag
- meanfloat
mean static magnitude
- zfloat
redshift
- tuple
single-filter light curve of shape [n_points, 1]