src.superphot_plus.model.data

Module Contents

Classes

ZtfData	Holds raw ZTF object data.
TrainData	Holds train and validation datasets.
TestData	Holds information about testing data.

class ZtfData

Holds raw ZTF object data.

names: List[int]

labels: List[int]

redshifts: List[int]

__iter__()

class TrainData

Holds train and validation datasets.

train_dataset: torch.utils.data.TensorDataset

val_dataset: torch.utils.data.TensorDataset

__iter__()

class TestData

Holds information about testing data.

test_features: numpy.ndarray

test_classes: numpy.ndarray

test_names: numpy.ndarray

test_group_idxs: List[int]

__iter__()