Index

_

__iter__() (TestData method) (TrainData method) (ZtfData method) __post_init__() (CurvePriors method) (MultibandPriors method) (Survey method)

_reindex() (Lightcurve method) _svi_helper_no_recompile() (in module numpyro_sampler)

A

add_snr_to_prob_csv() (in module utils) add_stamp_column() (in module src.superphot_plus.data_generation.alerce) add_to_new_csv() (in module src.superphot_plus.import_utils) adjust_log_dists() (in module src.superphot_plus.utils)

alerce (in module src.superphot_plus.data_generation.alerce) all_classes() (SupernovaClass class method) amp (CurvePriors attribute) append() (ModelMetrics method) aux_bands (MultibandPriors property)

B

band_as_int() (Lightcurve method) band_order (MultibandPriors attribute) bands (MultibandPriors attribute) BATCH_SIZE (in module src.superphot_plus.constants)

batch_size (ModelConfig attribute) best_val_loss (ModelConfig attribute) beta (CurvePriors attribute) BIGGER_SIZE (in module src.superphot_plus.constants)

C

calc_accuracy() (in module src.superphot_plus.utils) calculate_accuracy() (in module src.superphot_plus.utils) calculate_log_likelihood() (in module src.superphot_plus.utils) calculate_mse() (in module src.superphot_plus.utils) calculate_neg_chi_squareds() (in module src.superphot_plus.utils) canonicalize() (ElasticcClass class method) (SupernovaClass class method) classifier_results module classify_from_fit_params() (SuperphotClassifier method) CLASSIFY_LOG_FILE (in module src.superphot_plus.file_paths) classify_single_light_curve() (SuperphotClassifier method) clip_a (PriorFields attribute) clip_b (PriorFields attribute) clip_lightcurve_end() (in module src.superphot_plus.import_utils) CM_FOLDER (in module src.superphot_plus.file_paths)

compare_four_class_confusion_matrices() (in module confusion_matrices) compare_mag_distributions() (in module classifier_results) compare_oversampling() (in module sampling_results) confusion_matrices module convert_mags_to_flux() (in module src.superphot_plus.utils) create() (SuperphotClassifier class method) create_alerce_pred_csv() (in module utils) create_clean_models() (in module src.superphot_plus.data_generation.make_fake_spp_data) create_dataset() (in module src.superphot_plus.utils) create_jax_guide() (in module numpyro_sampler) create_jax_model() (in module numpyro_sampler) create_output_dirs() (TrainerBase method) create_prior() (in module src.superphot_plus.data_generation.make_fake_spp_data) create_ztf_model() (in module src.superphot_plus.data_generation.make_fake_spp_data) curr_epoch (ModelMetrics attribute) CurvePriors (class in fitting_priors)

D

DATA_DIR (in module src.superphot_plus.file_paths) DATA_DIRS (in module src.superphot_plus.file_paths) debug_string() (Lightcurve method) DEFAULT_MAX_FLUX (in module src.superphot_plus.data_generation.make_fake_spp_data) device (ModelConfig attribute)

DLOGZ (in module src.superphot_plus.constants) dust_filepath (in module src.superphot_plus.sfd) dynesty_sampler module DynestySampler (class in dynesty_sampler)

E

ElasticcClass (class in src.superphot_plus.elasticc_class) epoch_mins (ModelMetrics attribute) epoch_secs (ModelMetrics attribute) epoch_time() (in module src.superphot_plus.utils) EPOCHS (in module src.superphot_plus.constants)

evaluate() (SuperphotClassifier method) (SuperphotTrainer method) evaluate_epoch() (SuperphotClassifier method) extra_sigma (CurvePriors attribute) extract_wrong_classifications() (in module src.superphot_plus.utils)

F

f1_score() (in module src.superphot_plus.utils) filter_by_band() (Lightcurve method) (MultibandPriors method) find_max_flux() (Lightcurve method) FIT_PLOTS_FOLDER (in module src.superphot_plus.file_paths) FITS_DIR (in module src.superphot_plus.file_paths) fitting_priors module

flux_model() (in module src.superphot_plus.utils) format_params module forward() (SuperphotClassifier method) from_alerce_to_superphot_format() (SupernovaClass class method) from_file() (Lightcurve class method) (ModelConfig class method) (PosteriorSamples class method) (Survey class method)

G

gamma (CurvePriors attribute) gaussian() (in module utils) generate_files_from_antares() (in module src.superphot_plus.data_generation.antares) generate_flux_files() (in module src.superphot_plus.data_generation.alerce) generate_hp_sample() (SuperphotTuner method) generate_K_fold() (in module src.superphot_plus.format_data_ztf) generate_roc_curve() (in module classifier_results) generate_single_flux_file() (in module src.superphot_plus.data_generation.alerce) generate_test_data() (TrainerBase method) generate_train_data() (TrainerBase method) get_alerce_pred_class() (in module utils) get_all_unclassified_samples() (in module src.superphot_plus.data_generation.alerce) get_alternative_namings() (ElasticcClass class method) (SupernovaClass class method) get_band_extinctions() (in module src.superphot_plus.utils) get_band_extinctions_from_mwebv() (in module src.superphot_plus.utils)

get_classes_from_labels() (SupernovaClass class method) get_extinctions() (Survey method) get_labels_from_classes() (SupernovaClass class method) get_multiple_posterior_samples() (in module src.superphot_plus.file_utils) get_numpyro_cube() (in module src.superphot_plus.utils) get_ordered_wavelengths() (Survey method) get_posterior_filename() (in module src.superphot_plus.file_utils) get_posterior_samples() (in module src.superphot_plus.file_utils) get_predictions() (SuperphotClassifier method) get_predictions_from_fit_params() (SuperphotClassifier method) get_session_metrics() (in module src.superphot_plus.utils) get_survey_fracs() (in module utils) get_type_maps() (ElasticcClass class method) (SupernovaClass class method) get_values() (ModelMetrics method) goal_per_class (ModelConfig attribute)

H

has_posterior_samples() (in module src.superphot_plus.file_utils)

HIDDEN_DROPOUT_FRAC (in module src.superphot_plus.constants) histedges_equalN() (in module utils)

I

import_labels_only() (in module src.superphot_plus.format_data_ztf) import_lc() (in module src.superphot_plus.import_utils) import_snana() (in module src.superphot_plus.data_generation.snana)

INPUT_CSVS (in module src.superphot_plus.file_paths) input_dim (ModelConfig attribute) INPUT_DROPOUT_FRAC (in module src.superphot_plus.constants)

L

labels (ZtfData attribute) lax_helper_function() (in module numpyro_sampler) LEARNING_RATE (in module src.superphot_plus.constants) learning_rate (ModelConfig attribute) Lightcurve (class in src.superphot_plus.lightcurve)

lightcurves module lighten_color() (in module utils) load() (SuperphotClassifier class method) log_metrics_to_tensorboard() (in module src.superphot_plus.utils) LSST() (Survey class method)

M

MAX_ITER (in module src.superphot_plus.constants) mean (PriorFields attribute) MEANS_TRAINED_MODEL (in module src.superphot_plus.constants) MEDIUM_SIZE (in module src.superphot_plus.constants) METRICS_DIR (in module src.superphot_plus.file_paths) MIN_PER_FILTER (in module src.superphot_plus.data_generation.alerce) ModelConfig (class in src.superphot_plus.model.config) ModelMetrics (class in src.superphot_plus.model.metrics) MODELS_DIR (in module src.superphot_plus.file_paths) module classifier_results confusion_matrices dynesty_sampler fitting_priors format_params lightcurves numpyro_sampler sampler sampling_results src src.superphot_plus src.superphot_plus.constants src.superphot_plus.data_generation src.superphot_plus.data_generation.alerce src.superphot_plus.data_generation.antares src.superphot_plus.data_generation.make_fake_spp_data src.superphot_plus.data_generation.snana src.superphot_plus.elasticc_class src.superphot_plus.file_paths src.superphot_plus.file_utils src.superphot_plus.format_data_ztf src.superphot_plus.import_utils src.superphot_plus.lightcurve src.superphot_plus.model src.superphot_plus.model.classifier src.superphot_plus.model.config src.superphot_plus.model.data src.superphot_plus.model.metrics src.superphot_plus.posterior_samples src.superphot_plus.sfd src.superphot_plus.supernova_class src.superphot_plus.trainer src.superphot_plus.trainer_base src.superphot_plus.tuner src.superphot_plus.utils surveys utils

MultibandPriors (class in fitting_priors)

N

name (Survey attribute) names (ZtfData attribute) neurons_per_layer (ModelConfig attribute) NLIVE (in module src.superphot_plus.constants) normalization_means (ModelConfig attribute) normalization_stddevs (ModelConfig attribute) normalize_features() (in module src.superphot_plus.format_data_ztf)

num_epochs (ModelConfig attribute) NUM_FOLDS (in module src.superphot_plus.constants) num_folds (ModelConfig attribute) num_hidden_layers (ModelConfig attribute) numpyro_sampler module NumpyroSampler (class in numpyro_sampler)

O

obs_count() (Lightcurve method) ordered_bands (MultibandPriors property) output_dim (ModelConfig attribute)

OVERSAMPLE_SIZE (in module sampling_results) oversample_smote() (in module src.superphot_plus.format_data_ztf) oversample_using_posteriors() (in module src.superphot_plus.format_data_ztf)

P

pad_bands() (Lightcurve method) PAD_SIZE (in module src.superphot_plus.constants) param_labels() (in module format_params) params_valid() (in module src.superphot_plus.utils) plot_agreement_matrix_from_arrs() (in module confusion_matrices) plot_binary_confusion_matrix() (in module confusion_matrices) plot_chisquared_vs_accuracy() (in module classifier_results) plot_class_fractions() (in module classifier_results) plot_combined_posterior_space() (in module sampling_results) plot_confusion_matrix() (in module confusion_matrices) plot_corner_plot_all() (in module sampling_results) plot_expected_agreement_matrix() (in module confusion_matrices) plot_high_confidence_confusion_matrix() (in module confusion_matrices) plot_lc_fit() (in module lightcurves) plot_lightcurve_clipping() (in module lightcurves) plot_matrices() (in module confusion_matrices) plot_model_metrics() (in module classifier_results) plot_oversampling_1d() (in module sampling_results) plot_param_distributions() (in module sampling_results) plot_phase_vs_accuracy() (in module classifier_results) plot_posterior_hist_numpyro_dict() (in module sampling_results) plot_redshifts_abs_mags() (in module classifier_results) plot_sampling_lc_fit() (in module lightcurves) plot_sampling_lc_fit_numpyro() (in module lightcurves)

plot_sampling_trace_numpyro() (in module sampling_results) plot_snr_hist() (in module classifier_results) plot_snr_npoints_vs_accuracy() (in module classifier_results) plot_true_agreement_matrix() (in module confusion_matrices) PosteriorSamples (class in src.superphot_plus.posterior_samples) print_last() (ModelMetrics method) PRIOR_A (in module src.superphot_plus.data_generation.make_fake_spp_data) PRIOR_A_g (in module src.superphot_plus.data_generation.make_fake_spp_data) PRIOR_BETA (in module src.superphot_plus.data_generation.make_fake_spp_data) PRIOR_BETA_g (in module src.superphot_plus.data_generation.make_fake_spp_data) PRIOR_EXTRA_SIGMA (in module src.superphot_plus.data_generation.make_fake_spp_data) PRIOR_EXTRA_SIGMA_g (in module src.superphot_plus.data_generation.make_fake_spp_data) PRIOR_GAMMA (in module src.superphot_plus.data_generation.make_fake_spp_data) PRIOR_GAMMA_g (in module src.superphot_plus.data_generation.make_fake_spp_data) prior_helper() (in module numpyro_sampler) PRIOR_T0 (in module src.superphot_plus.data_generation.make_fake_spp_data) PRIOR_T0_g (in module src.superphot_plus.data_generation.make_fake_spp_data) PRIOR_TAU_FALL (in module src.superphot_plus.data_generation.make_fake_spp_data) PRIOR_TAU_FALL_g (in module src.superphot_plus.data_generation.make_fake_spp_data) PRIOR_TAU_RISE (in module src.superphot_plus.data_generation.make_fake_spp_data) PRIOR_TAU_RISE_g (in module src.superphot_plus.data_generation.make_fake_spp_data) PriorFields (class in fitting_priors) priors (Survey attribute) PROBS_FILE (in module src.superphot_plus.file_paths) PROBS_FILE2 (in module src.superphot_plus.file_paths)

R

read_probs_csv() (in module utils) redshifts (ZtfData attribute) reference_band (MultibandPriors attribute) retrieve_four_class_info() (in module utils) return_new_classifications() (SuperphotClassifier method) run() (SuperphotTrainer method) (SuperphotTuner method) run_cross_validation() (SuperphotTuner method)

run_mcmc() (in module dynesty_sampler) (in module numpyro_sampler) run_multi_curve() (DynestySampler method) (NumpyroSampler method) (Sampler method) run_single_curve() (DynestySampler method) (NumpyroSampler method) (Sampler method)

S

sample_mean() (PosteriorSamples method) sampler module Sampler (class in sampler) sampling_results module save() (SuperphotClassifier method) save_class_fractions() (in module classifier_results) save_test_probabilities() (in module src.superphot_plus.utils) save_to_file() (Lightcurve method) (PosteriorSamples method) SEED (in module src.superphot_plus.constants) set_best_val_loss() (ModelConfig method) set_global_plot_formatting() (in module format_params) set_non_tunable_params() (ModelConfig method) setup_model() (SuperphotTrainer method) SMALL_SIZE (in module src.superphot_plus.constants) sort_by_time() (Lightcurve method) split_train_test() (TrainerBase method) src module src.superphot_plus module src.superphot_plus.constants module src.superphot_plus.data_generation module src.superphot_plus.data_generation.alerce module src.superphot_plus.data_generation.antares module src.superphot_plus.data_generation.make_fake_spp_data module src.superphot_plus.data_generation.snana module src.superphot_plus.elasticc_class module src.superphot_plus.file_paths module src.superphot_plus.file_utils module src.superphot_plus.format_data_ztf module src.superphot_plus.import_utils module

src.superphot_plus.lightcurve module src.superphot_plus.model module src.superphot_plus.model.classifier module src.superphot_plus.model.config module src.superphot_plus.model.data module src.superphot_plus.model.metrics module src.superphot_plus.posterior_samples module src.superphot_plus.sfd module src.superphot_plus.supernova_class module src.superphot_plus.trainer module src.superphot_plus.trainer_base module src.superphot_plus.tuner module src.superphot_plus.utils module std (PriorFields attribute) STDDEVS_TRAINED_MODEL (in module src.superphot_plus.constants) SUPERLUMINOUS_SUPERNOVA_I (ElasticcClass attribute) (SupernovaClass attribute) SUPERNOVA_IA (ElasticcClass attribute) (SupernovaClass attribute) SUPERNOVA_IBC (ElasticcClass attribute) (SupernovaClass attribute) SUPERNOVA_II (ElasticcClass attribute) (SupernovaClass attribute) SUPERNOVA_IIN (ElasticcClass attribute) (SupernovaClass attribute) SupernovaClass (class in src.superphot_plus.supernova_class) SuperphotClassifier (class in src.superphot_plus.model.classifier) SuperphotTrainer (class in src.superphot_plus.trainer) SuperphotTuner (class in src.superphot_plus.tuner) Survey (class in surveys) surveys module

T

t_0 (CurvePriors attribute) tally_each_class() (in module src.superphot_plus.format_data_ztf) tau_fall (CurvePriors attribute) tau_rise (CurvePriors attribute) test_classes (TestData attribute) test_features (TestData attribute) test_group_idxs (TestData attribute) test_names (TestData attribute) TestData (class in src.superphot_plus.model.data) to_numpy() (CurvePriors method) (MultibandPriors method) (PriorFields method) train() (SuperphotTrainer method)

train_acc (ModelMetrics attribute) train_and_validate() (SuperphotClassifier method) train_dataset (TrainData attribute) train_epoch() (SuperphotClassifier method) train_loss (ModelMetrics attribute) TrainData (class in src.superphot_plus.model.data) TRAINED_CONFIG_FN (in module src.superphot_plus.file_paths) TRAINED_MODEL_FN (in module src.superphot_plus.file_paths) TRAINED_MODEL_PARAMS (in module src.superphot_plus.constants) TrainerBase (class in src.superphot_plus.trainer_base) trunc_gauss() (in module src.superphot_plus.data_generation.make_fake_spp_data) trunc_norm() (in module numpyro_sampler) tune_model() (SuperphotTuner method)

U

unique_bands() (Lightcurve method)

utils module

V

val_acc (ModelMetrics attribute) val_dataset (TrainData attribute)

val_loss (ModelMetrics attribute) VALID_RATIO (in module src.superphot_plus.constants)

W

wavelengths (Survey attribute) write_metrics_to_file() (in module src.superphot_plus.utils)

write_to_file() (ModelConfig method) (Survey method) WRONGLY_CLASSIFIED_FOLDER (in module src.superphot_plus.file_paths)

Z

ZTF() (Survey class method)

ztf_noise_model() (in module src.superphot_plus.data_generation.make_fake_spp_data) ZtfData (class in src.superphot_plus.model.data)