oracle.pretrained.ELAsTiCC

Pretrained model(s) for the ELAsTiCC dataset.

Functions

augment_table(table)

Augments an astronomical observation table by cleaning and transforming its data. This function performs several modifications on the input table: - Creates a copy of the table to avoid modifying the original. - Removes rows where the 'PHOTFLAG' column indicates saturation (using a bitmask with 1024). - Reassigns the 'PHOTFLAG' column values: * Sets to 1 for detections (when the bitmask with 4096 is non-zero). * Sets to 0 for non-detections. - Converts band labels in the 'BAND' column to mean wavelengths using the 'LSST_passband_to_wavelengths' mapping. - Normalizes time data by subtracting the Modified Julian Date (MJD) of the first detection from all 'MJD' entries. - Reorders the columns based on the predefined list 'time_dependent_feature_list'. - Iterates over the metadata ('meta') of the table and replaces any values that match entries in 'missing_data_flags' with 'flag_value'.

Classes

ORACLE1_ELAsTiCC([model_dir])

ORACLE1_ELAsTiCC is a model class that inherits from GRU_MD designed to load a pre-trained ELAsTiCC model and perform predictions on time series data augmented with static features.

ORACLE1_ELAsTiCC_lite([model_dir])

Predict class probability scores for a single time-series table.