oracle package

Subpackages

• oracle.custom_datasets package
  • Submodules
    • oracle.custom_datasets.BTS module
      • BTS_LC_Dataset
      • custom_collate_BTS()
      • show_batch()
      • truncate_BTS_light_curve_by_days_since_trigger()
    • oracle.custom_datasets.ELAsTiCC module
      • ELAsTiCC_LC_Dataset
      • custom_collate_ELAsTiCC()
      • show_batch()
      • truncate_ELAsTiCC_light_curve_by_days_since_trigger()
      • truncate_ELAsTiCC_light_curve_fractionally()
    • oracle.custom_datasets.ZTF_sims module
      • ZTF_SIM_LC_Dataset
      • custom_collate_ZTF_SIM()
      • show_batch()
      • truncate_ZTF_SIM_light_curve_by_days_since_trigger()
      • truncate_ZTF_SIM_light_curve_fractionally()
  • Module contents
• oracle.pretrained package
  • Submodules
    • oracle.pretrained.BTS module
      • ORACLE1_BTS
      • augment_table()
    • oracle.pretrained.ELAsTiCC module
      • ORACLE1_ELAsTiCC
      • ORACLE1_ELAsTiCC_lite
      • augment_table()
  • Module contents
• oracle.utils package
  • Submodules
    • oracle.utils.ELAsTiCC_combine_class_parquet module
      • main()
      • parse_args()
    • oracle.utils.ELAsTiCC_fits_to_parquet module
      • main()
      • parse_args()
    • oracle.utils.ZTF_sim_combine_class_parquet module
      • main()
      • parse_args()
    • oracle.utils.ZTF_sim_fits_to_parquet module
      • main()
      • parse_args()
    • oracle.utils.bts_csv_to_parquet module
      • main()
      • parse_args()
    • oracle.utils.split_parquet module
      • parse_args()
      • split_parquet()
  • Module contents

Submodules

• oracle.architectures module
  • GRU
    • GRU.forward()
    • GRU.get_latent_space_embeddings()
  • GRU_MD
    • GRU_MD.forward()
    • GRU_MD.get_latent_space_embeddings()
  • GRU_MD_MM
    • GRU_MD_MM.forward()
    • GRU_MD_MM.get_latent_space_embeddings()
  • Hierarchical_classifier
    • Hierarchical_classifier.embed()
    • Hierarchical_classifier.get_latent_space_embeddings()
    • Hierarchical_classifier.predict()
    • Hierarchical_classifier.predict_class_probabilities()
    • Hierarchical_classifier.predict_class_probabilities_df()
    • Hierarchical_classifier.predict_conditional_probabilities()
    • Hierarchical_classifier.predict_conditional_probabilities_df()
    • Hierarchical_classifier.predict_full_scores()
    • Hierarchical_classifier.score()
• oracle.constants module
• oracle.loss module
  • WHXE_Loss
    • WHXE_Loss.compute_lambda_term()
    • WHXE_Loss.forward()
    • WHXE_Loss.get_class_weights()
• oracle.presets module
  • get_class_weights()
  • get_model()
  • get_test_loaders()
  • get_train_loader()
  • get_val_loader()
  • worker_init_fn()
• oracle.taxonomies module
  • BTS_Taxonomy
  • ORACLE_Taxonomy
  • Taxonomy
    • Taxonomy.get_class_probabilities()
    • Taxonomy.get_conditional_probabilities()
    • Taxonomy.get_depths()
    • Taxonomy.get_hierarchical_one_hot_encoding()
    • Taxonomy.get_leaf_nodes()
    • Taxonomy.get_level_order_traversal()
    • Taxonomy.get_nodes_by_depth()
    • Taxonomy.get_parent_nodes()
    • Taxonomy.get_paths()
    • Taxonomy.get_sibling_masks()
    • Taxonomy.plot_colored_taxonomy()
    • Taxonomy.plot_taxonomy()
• oracle.test module
  • main()
  • parse_args()
  • run_testing_loop()
• oracle.tester module
  • Tester
    • Tester.create_classification_report()
    • Tester.create_loss_history_plot()
    • Tester.create_metric_phase_plots()
    • Tester.get_metric_over_all_phases()
    • Tester.make_embeddings_for_AD()
    • Tester.merge_performance_tables()
    • Tester.run_all_analysis()
    • Tester.setup_testing()
• oracle.train module
  • get_wandb_run()
  • main()
  • parse_args()
  • run_training_loop()
  • save_args_to_csv()
• oracle.trainer module
  • EarlyStopper
    • EarlyStopper.patience
    • EarlyStopper.min_delta
    • EarlyStopper.counter
    • EarlyStopper.min_validation_loss
    • EarlyStopper.early_stop()
  • Trainer
    • Trainer.fit()
    • Trainer.log_data_in_wandb()
    • Trainer.save_loss_history()
    • Trainer.save_model_in_wandb()
    • Trainer.setup_training()
    • Trainer.train_one_epoch()
    • Trainer.validate_one_epoch()
• oracle.visualization module
  • plot_average_performance_over_all_phases()
  • plot_class_wise_performance_over_all_phases()
  • plot_confusion_matrix()
  • plot_plain_cf()
  • plot_roc_curves()
  • plot_train_val_history()
  • plot_umap()

Module contents

ORACLE is a general framework for hierarchical classification.