oracle.custom_datasets.ZTF_sims

Custom dataset class for the simulated ZTF light curve dataset.

Functions

custom_collate_ZTF_SIM(batch)	Collates a batch of ZTF simulation samples into a single dictionary of tensors suitable for model input.
show_batch(images, labels[, n])	Display a grid of images with corresponding labels.
truncate_ZTF_SIM_light_curve_by_days_since_trigger(x_ts, d)	Truncates a ZTF SIM light curve by retaining only the observations within a specified number of days since the trigger (first detection).
truncate_ZTF_SIM_light_curve_fractionally(x_ts)	Truncate a ZTF simulation light curve by retaining only a fraction of its observations.

Classes

ZTF_SIM_LC_Dataset(parquet_file_path[, ...])