Index

A

augment_table() (in module oracle.pretrained.BTS) (in module oracle.pretrained.ELAsTiCC), [1]

B

BTS_LC_Dataset (class in oracle.custom_datasets.BTS), [1]

BTS_Taxonomy (class in oracle.taxonomies), [1]

C

clean_up_dataset() (oracle.custom_datasets.BTS.BTS_LC_Dataset method), [1] (oracle.custom_datasets.ELAsTiCC.ELAsTiCC_LC_Dataset method), [1] (oracle.custom_datasets.ZTF_sims.ZTF_SIM_LC_Dataset method), [1] compute_lambda_term() (oracle.loss.WHXE_Loss method), [1] convert_mags_to_flux() (oracle.custom_datasets.BTS.BTS_LC_Dataset method), [1] counter (oracle.trainer.EarlyStopper attribute), [1]

create_classification_report() (oracle.tester.Tester method), [1] create_loss_history_plot() (oracle.tester.Tester method), [1] create_metric_phase_plots() (oracle.tester.Tester method), [1] custom_collate_BTS() (in module oracle.custom_datasets.BTS), [1] custom_collate_ELAsTiCC() (in module oracle.custom_datasets.ELAsTiCC), [1] custom_collate_ZTF_SIM() (in module oracle.custom_datasets.ZTF_sims), [1]

E

early_stop() (oracle.trainer.EarlyStopper method), [1] EarlyStopper (class in oracle.trainer), [1] ELAsTiCC_LC_Dataset (class in oracle.custom_datasets.ELAsTiCC), [1] embed() (oracle.architectures.Hierarchical_classifier method), [1] (oracle.pretrained.BTS.ORACLE1_BTS method) (oracle.pretrained.ELAsTiCC.ORACLE1_ELAsTiCC method), [1] (oracle.pretrained.ELAsTiCC.ORACLE1_ELAsTiCC_lite method), [1]

exclude_classes() (oracle.custom_datasets.BTS.BTS_LC_Dataset method), [1] (oracle.custom_datasets.ELAsTiCC.ELAsTiCC_LC_Dataset method), [1]

F

fit() (oracle.trainer.Trainer method), [1] forward() (oracle.architectures.GRU method), [1] (oracle.architectures.GRU_MD method), [1] (oracle.architectures.GRU_MD_MM method), [1] (oracle.loss.WHXE_Loss method), [1]

G

get_all_labels() (oracle.custom_datasets.BTS.BTS_LC_Dataset method), [1] (oracle.custom_datasets.ELAsTiCC.ELAsTiCC_LC_Dataset method), [1] (oracle.custom_datasets.ZTF_sims.ZTF_SIM_LC_Dataset method), [1] get_class_probabilities() (oracle.taxonomies.Taxonomy method), [1] get_class_weights() (in module oracle.presets), [1] (oracle.loss.WHXE_Loss method), [1] get_conditional_probabilities() (oracle.taxonomies.Taxonomy method), [1] get_depths() (oracle.taxonomies.Taxonomy method), [1] get_hierarchical_one_hot_encoding() (oracle.taxonomies.Taxonomy method), [1] get_latent_space_embeddings() (oracle.architectures.GRU method), [1] (oracle.architectures.GRU_MD method), [1] (oracle.architectures.GRU_MD_MM method), [1] (oracle.architectures.Hierarchical_classifier method), [1] get_lc_plots() (oracle.custom_datasets.BTS.BTS_LC_Dataset method), [1] (oracle.custom_datasets.ELAsTiCC.ELAsTiCC_LC_Dataset method), [1] (oracle.custom_datasets.ZTF_sims.ZTF_SIM_LC_Dataset method), [1]

get_leaf_nodes() (oracle.taxonomies.Taxonomy method), [1] get_level_order_traversal() (oracle.taxonomies.Taxonomy method), [1] get_metric_over_all_phases() (oracle.tester.Tester method), [1] get_model() (in module oracle.presets), [1] get_nodes_by_depth() (oracle.taxonomies.Taxonomy method), [1] get_parent_nodes() (oracle.taxonomies.Taxonomy method), [1] get_paths() (oracle.taxonomies.Taxonomy method), [1] get_postage_stamp() (oracle.custom_datasets.BTS.BTS_LC_Dataset method), [1] get_postage_stamp_plot() (oracle.custom_datasets.BTS.BTS_LC_Dataset method), [1] get_sibling_masks() (oracle.taxonomies.Taxonomy method), [1] get_test_loaders() (in module oracle.presets), [1] get_train_loader() (in module oracle.presets), [1] get_val_loader() (in module oracle.presets), [1] get_wandb_run() (in module oracle.train), [1] GRU (class in oracle.architectures), [1] GRU_MD (class in oracle.architectures), [1] GRU_MD_MM (class in oracle.architectures), [1]

H

Hierarchical_classifier (class in oracle.architectures), [1]

L

limit_max_samples_per_class() (oracle.custom_datasets.BTS.BTS_LC_Dataset method), [1] (oracle.custom_datasets.ELAsTiCC.ELAsTiCC_LC_Dataset method), [1] (oracle.custom_datasets.ZTF_sims.ZTF_SIM_LC_Dataset method), [1]

log_data_in_wandb() (oracle.trainer.Trainer method), [1]

M

main() (in module oracle.test), [1] (in module oracle.train), [1] (in module oracle.utils.bts_csv_to_parquet), [1] (in module oracle.utils.ELAsTiCC_combine_class_parquet), [1] (in module oracle.utils.ELAsTiCC_fits_to_parquet), [1] (in module oracle.utils.ZTF_sim_combine_class_parquet), [1] (in module oracle.utils.ZTF_sim_fits_to_parquet), [1] make_batch() (oracle.pretrained.BTS.ORACLE1_BTS method) (oracle.pretrained.ELAsTiCC.ORACLE1_ELAsTiCC method), [1] (oracle.pretrained.ELAsTiCC.ORACLE1_ELAsTiCC_lite method), [1] make_embeddings_for_AD() (oracle.tester.Tester method), [1] map_models_to_classes() (oracle.custom_datasets.ELAsTiCC.ELAsTiCC_LC_Dataset method), [1] merge_performance_tables() (oracle.tester.Tester method), [1] min_delta (oracle.trainer.EarlyStopper attribute), [1] min_validation_loss (oracle.trainer.EarlyStopper attribute), [1] model_dir (oracle.pretrained.BTS.ORACLE1_BTS attribute) (oracle.pretrained.ELAsTiCC.ORACLE1_ELAsTiCC attribute), [1] module oracle, [1], [2] oracle.architectures, [1], [2] oracle.constants, [1], [2] oracle.custom_datasets, [1], [2] oracle.custom_datasets.BTS, [1], [2] oracle.custom_datasets.ELAsTiCC, [1], [2] oracle.custom_datasets.ZTF_sims, [1], [2] oracle.loss, [1], [2] oracle.presets, [1], [2] oracle.pretrained, [1], [2] oracle.pretrained.BTS, [1] oracle.pretrained.ELAsTiCC, [1], [2] oracle.taxonomies, [1], [2] oracle.test, [1], [2] oracle.tester, [1], [2] oracle.train, [1], [2] oracle.trainer, [1], [2] oracle.utils, [1], [2] oracle.utils.bts_csv_to_parquet, [1], [2] oracle.utils.ELAsTiCC_combine_class_parquet, [1], [2] oracle.utils.ELAsTiCC_fits_to_parquet, [1], [2] oracle.utils.split_parquet, [1], [2] oracle.utils.ZTF_sim_combine_class_parquet, [1], [2] oracle.utils.ZTF_sim_fits_to_parquet, [1], [2] oracle.visualization, [1], [2]

O

oracle module, [1], [2] oracle.architectures module, [1], [2] oracle.constants module, [1], [2] oracle.custom_datasets module, [1], [2] oracle.custom_datasets.BTS module, [1], [2] oracle.custom_datasets.ELAsTiCC module, [1], [2] oracle.custom_datasets.ZTF_sims module, [1], [2] oracle.loss module, [1], [2] oracle.presets module, [1], [2] oracle.pretrained module, [1], [2] oracle.pretrained.BTS module, [1] oracle.pretrained.ELAsTiCC module, [1], [2] oracle.taxonomies module, [1], [2] oracle.test module, [1], [2]

oracle.tester module, [1], [2] oracle.train module, [1], [2] oracle.trainer module, [1], [2] oracle.utils module, [1], [2] oracle.utils.bts_csv_to_parquet module, [1], [2] oracle.utils.ELAsTiCC_combine_class_parquet module, [1], [2] oracle.utils.ELAsTiCC_fits_to_parquet module, [1], [2] oracle.utils.split_parquet module, [1], [2] oracle.utils.ZTF_sim_combine_class_parquet module, [1], [2] oracle.utils.ZTF_sim_fits_to_parquet module, [1], [2] oracle.visualization module, [1], [2] ORACLE1_BTS (class in oracle.pretrained.BTS) ORACLE1_ELAsTiCC (class in oracle.pretrained.ELAsTiCC), [1] ORACLE1_ELAsTiCC_lite (class in oracle.pretrained.ELAsTiCC), [1] ORACLE_Taxonomy (class in oracle.taxonomies), [1] over_sample_minority_classes() (oracle.custom_datasets.BTS.BTS_LC_Dataset method), [1]

P

parse_args() (in module oracle.test), [1] (in module oracle.train), [1] (in module oracle.utils.bts_csv_to_parquet), [1] (in module oracle.utils.ELAsTiCC_combine_class_parquet), [1] (in module oracle.utils.ELAsTiCC_fits_to_parquet), [1] (in module oracle.utils.split_parquet), [1] (in module oracle.utils.ZTF_sim_combine_class_parquet), [1] (in module oracle.utils.ZTF_sim_fits_to_parquet), [1] patience (oracle.trainer.EarlyStopper attribute), [1] plot_average_performance_over_all_phases() (in module oracle.visualization), [1] plot_class_wise_performance_over_all_phases() (in module oracle.visualization), [1] plot_colored_taxonomy() (oracle.taxonomies.Taxonomy method), [1] plot_confusion_matrix() (in module oracle.visualization), [1] plot_plain_cf() (in module oracle.visualization), [1] plot_roc_curves() (in module oracle.visualization), [1] plot_taxonomy() (oracle.taxonomies.Taxonomy method), [1]

plot_train_val_history() (in module oracle.visualization), [1] plot_umap() (in module oracle.visualization), [1] predict() (oracle.architectures.Hierarchical_classifier method), [1] (oracle.pretrained.BTS.ORACLE1_BTS method) (oracle.pretrained.ELAsTiCC.ORACLE1_ELAsTiCC method), [1] (oracle.pretrained.ELAsTiCC.ORACLE1_ELAsTiCC_lite method), [1] predict_class_probabilities() (oracle.architectures.Hierarchical_classifier method), [1] predict_class_probabilities_df() (oracle.architectures.Hierarchical_classifier method), [1] predict_conditional_probabilities() (oracle.architectures.Hierarchical_classifier method), [1] predict_conditional_probabilities_df() (oracle.architectures.Hierarchical_classifier method), [1] predict_full_scores() (oracle.architectures.Hierarchical_classifier method), [1] (oracle.pretrained.BTS.ORACLE1_BTS method) (oracle.pretrained.ELAsTiCC.ORACLE1_ELAsTiCC method), [1] (oracle.pretrained.ELAsTiCC.ORACLE1_ELAsTiCC_lite method), [1] print_dataset_composition() (oracle.custom_datasets.BTS.BTS_LC_Dataset method), [1] (oracle.custom_datasets.ZTF_sims.ZTF_SIM_LC_Dataset method), [1]

R

run_all_analysis() (oracle.tester.Tester method), [1]

run_testing_loop() (in module oracle.test), [1] run_training_loop() (in module oracle.train), [1]

S

save_args_to_csv() (in module oracle.train), [1] save_loss_history() (oracle.trainer.Trainer method), [1] save_model_in_wandb() (oracle.trainer.Trainer method), [1] score() (oracle.architectures.Hierarchical_classifier method), [1] (oracle.pretrained.BTS.ORACLE1_BTS method) (oracle.pretrained.ELAsTiCC.ORACLE1_ELAsTiCC method), [1] (oracle.pretrained.ELAsTiCC.ORACLE1_ELAsTiCC_lite method), [1]

setup_testing() (oracle.tester.Tester method), [1] setup_training() (oracle.trainer.Trainer method), [1] show_batch() (in module oracle.custom_datasets.BTS), [1] (in module oracle.custom_datasets.ELAsTiCC), [1] (in module oracle.custom_datasets.ZTF_sims), [1] split_parquet() (in module oracle.utils.split_parquet), [1] static_feature_dim (oracle.pretrained.BTS.ORACLE1_BTS attribute) (oracle.pretrained.ELAsTiCC.ORACLE1_ELAsTiCC attribute), [1]

T

Taxonomy (class in oracle.taxonomies), [1] taxonomy (oracle.pretrained.BTS.ORACLE1_BTS attribute) (oracle.pretrained.ELAsTiCC.ORACLE1_ELAsTiCC attribute), [1] Tester (class in oracle.tester), [1] train_one_epoch() (oracle.trainer.Trainer method), [1] Trainer (class in oracle.trainer), [1]

truncate_BTS_light_curve_by_days_since_trigger() (in module oracle.custom_datasets.BTS), [1] truncate_ELAsTiCC_light_curve_by_days_since_trigger() (in module oracle.custom_datasets.ELAsTiCC), [1] truncate_ELAsTiCC_light_curve_fractionally() (in module oracle.custom_datasets.ELAsTiCC), [1] truncate_ZTF_SIM_light_curve_by_days_since_trigger() (in module oracle.custom_datasets.ZTF_sims), [1] truncate_ZTF_SIM_light_curve_fractionally() (in module oracle.custom_datasets.ZTF_sims), [1] ts_feature_dim (oracle.pretrained.BTS.ORACLE1_BTS attribute) (oracle.pretrained.ELAsTiCC.ORACLE1_ELAsTiCC attribute), [1]

V

validate_one_epoch() (oracle.trainer.Trainer method), [1]

W

WHXE_Loss (class in oracle.loss), [1]

worker_init_fn() (in module oracle.presets), [1]

Z

ZTF_SIM_LC_Dataset (class in oracle.custom_datasets.ZTF_sims), [1]