Index

_

__iter__() (PosteriorSamplesGroup method) (TestData method), [1] (TrainData method), [1] (ZtfData method) __post_init__() (CurvePriors method) (MultibandPriors method) (PosteriorSamplesGroup method) (SuperphotConfig method) (Survey method) (TestData method)

_create_model_instance() (SuperphotTrainer method) _load_model_instance() (SuperphotTrainer method) _reindex() (Lightcurve method) _svi_helper_no_recompile() (in module src.superphot_plus.samplers.numpyro_sampler)

A

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

all_classes() (SupernovaClass class method) amp (CurvePriors attribute) append() (ModelMetrics method), [1] args (in module scripts.generate_fits) aux_bands (MultibandPriors property)

B

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

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

C

calc_accuracy() (in module src.superphot_plus.utils) calc_calibration_curve() (in module src.superphot_plus.plotting.utils) calc_precision_recall() (in module src.superphot_plus.plotting.utils) calculate_accuracy() (in module src.superphot_plus.utils) calculate_chi_squareds() (in module src.superphot_plus.utils) calculate_log_likelihood() (in module src.superphot_plus.utils) calculate_mse() (in module src.superphot_plus.utils) canonicalize() (ElasticcClass class method) (SupernovaClass class method) canonicalize_labels() (PosteriorSamplesGroup method) classify_from_fit_params() (SuperphotClassifier method) (SuperphotLightGBM method) (SuperphotMLP method) classify_single_light_curve() (SuperphotClassifier method) (SuperphotTrainer method) clip_a (PriorFields attribute) clip_b (PriorFields attribute) clip_lightcurve_end() (in module src.superphot_plus.import_utils) cm_dir (SuperphotConfig attribute)

compare_four_class_confusion_matrices() (in module src.superphot_plus.plotting.confusion_matrices) compare_mag_distributions() (in module src.superphot_plus.plotting.classifier_results) compare_oversampling() (in module src.superphot_plus.plotting.sampling_results) convert_mags_to_flux() (in module src.superphot_plus.utils) convert_to_lcdata_h5() (in module parsnip) create() (SuperphotClassifier class method) (SuperphotMLP class method) create_alerce_pred_csv() (in module src.superphot_plus.plotting.utils) create_clean_models() (in module src.superphot_plus.data_generation.make_fake_spp_data) create_dataset() (in module src.superphot_plus.utils) create_dirs (SuperphotConfig attribute) create_jax_guide() (in module src.superphot_plus.samplers.numpyro_sampler) create_jax_model() (in module src.superphot_plus.samplers.numpyro_sampler) create_metatable() (in module superphot) 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), [1] CurvePriors (class in src.superphot_plus.surveys.fitting_priors) CUSTOM_COLORSET (in module src.superphot_plus.plotting.format_params)

D

data_dir (SuperphotConfig attribute) debug_string() (Lightcurve method) DEFAULT_FEATURES (in module parsnip) DEFAULT_MAX_FLUX (in module src.superphot_plus.data_generation.make_fake_spp_data) DEVICE (in module parsnip)

device (ModelConfig attribute) (SuperphotConfig attribute) DLOGZ (in module src.superphot_plus.constants) dust_filepath (in module src.superphot_plus.sfd) DynestySampler (class in src.superphot_plus.samplers.dynesty_sampler)

E

ElasticcClass (class in src.superphot_plus.elasticc_class) encode_raenn_features() (in module superraenn) encode_superphot_features() (in module superphot) encode_with_parsnip() (in module parsnip) ENCODING_N_DEFAULT (in module superraenn) epoch_mins (ModelMetrics attribute), [1] epoch_secs (ModelMetrics attribute), [1] epoch_time() (in module src.superphot_plus.utils) EPOCHS (in module src.superphot_plus.constants)

evaluate() (SuperphotClassifier method) (SuperphotLightGBM method) (SuperphotMLP method) (SuperphotTrainer method) evaluate_epoch() (SuperphotClassifier method) (SuperphotMLP method) extra_sigma (CurvePriors attribute) extract_cmd_args() (in module scripts.generate_fits) extract_wrong_classifications() (in module src.superphot_plus.utils)

F

f1_score() (in module src.superphot_plus.utils) figs_dir (SuperphotConfig attribute) filter_by_band() (Lightcurve method) (MultibandPriors method) find_max_flux() (Lightcurve method) fit_lcs_superphot() (in module superphot) fit_plots_dir (SuperphotConfig attribute) fits_dir (SuperphotConfig attribute)

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

G

gamma (CurvePriors attribute) gaussian() (in module src.superphot_plus.plotting.utils) generate_data() (PosteriorsGenerator method) 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_roc_curve() (in module src.superphot_plus.plotting.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 src.superphot_plus.plotting.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_posteriors_fn() (PosteriorsGenerator method) get_posteriors_to_generate() (PosteriorsGenerator method) get_predictions() (SuperphotClassifier method) (SuperphotMLP method) get_predictions_from_fit_params() (SuperphotClassifier method) (SuperphotMLP method) get_session_metrics() (in module src.superphot_plus.utils) get_survey_fracs() (in module src.superphot_plus.plotting.utils) get_type_maps() (ElasticcClass class method) (SupernovaClass class method) get_values() (ModelMetrics method), [1] goal_per_class (ModelConfig attribute) (SuperphotConfig 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 src.superphot_plus.plotting.utils)

I

ignore_param_idxs (PosteriorSamplesGroup attribute) IminuitSampler (class in src.superphot_plus.samplers.iminuit_sampler) 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 (SuperphotConfig attribute) input_dim (ModelConfig attribute) (SuperphotConfig attribute) INPUT_DROPOUT_FRAC (in module src.superphot_plus.constants)

K

k_fold_split_train_test() (TrainerBase method)

L

labels (ZtfData attribute) lax_helper_function() (in module src.superphot_plus.samplers.numpyro_sampler) LEARNING_RATE (in module src.superphot_plus.constants) learning_rate (ModelConfig attribute) (SuperphotConfig attribute) LiCuSampler (class in src.superphot_plus.samplers.licu_sampler) Lightcurve (class in src.superphot_plus.lightcurve) lighten_color() (in module src.superphot_plus.plotting.utils)

load() (SuperphotClassifier class method) (SuperphotLightGBM class method) (SuperphotMLP class method) load_csv() (TrainerBase method) log_fn (SuperphotConfig attribute) log_metrics_to_tensorboard() (in module src.superphot_plus.utils) LOW_SNR_FILE (in module src.superphot_plus.import_utils) LOW_VAR_FILE (in module src.superphot_plus.import_utils) LSST() (Survey class method)

M

make_binary() (PosteriorSamplesGroup method) make_fully_redshift_independent() (PosteriorSamplesGroup method) 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 (SuperphotConfig attribute) MIN_PER_FILTER (in module src.superphot_plus.data_generation.alerce) ModelConfig (class in src.superphot_plus.sfd.model.config) ModelMetrics (class in src.superphot_plus.model.metrics) (class in src.superphot_plus.sfd.model.metrics) models_dir (SuperphotConfig attribute) module parsnip scripts scripts.generate_fits src src.superphot_plus src.superphot_plus.config 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_utils src.superphot_plus.format_data_ztf src.superphot_plus.import_utils src.superphot_plus.lightcurve src.superphot_plus.model src.superphot_plus.model.data src.superphot_plus.model.lightgbm src.superphot_plus.model.metrics src.superphot_plus.model.mlp src.superphot_plus.plotting src.superphot_plus.plotting.classifier_results src.superphot_plus.plotting.confusion_matrices src.superphot_plus.plotting.format_params src.superphot_plus.plotting.lightcurves src.superphot_plus.plotting.sampling_results src.superphot_plus.plotting.utils src.superphot_plus.posterior_samples src.superphot_plus.samplers src.superphot_plus.samplers.dynesty_sampler src.superphot_plus.samplers.iminuit_sampler src.superphot_plus.samplers.licu_sampler src.superphot_plus.samplers.numpyro_sampler src.superphot_plus.samplers.sampler src.superphot_plus.sfd src.superphot_plus.sfd.model src.superphot_plus.sfd.model.classifier src.superphot_plus.sfd.model.config src.superphot_plus.sfd.model.data src.superphot_plus.sfd.model.metrics src.superphot_plus.supernova_class src.superphot_plus.surveys src.superphot_plus.surveys.fitting_priors src.superphot_plus.surveys.surveys src.superphot_plus.trainer src.superphot_plus.trainer_base src.superphot_plus.tuner src.superphot_plus.utils superphot superraenn

MultibandPriors (class in src.superphot_plus.surveys.fitting_priors)

N

N_EPOCH_DEFAULT (in module superraenn) name (Survey attribute) names (ZtfData attribute) NEURON_N_DEFAULT (in module superraenn) neurons_per_layer (ModelConfig attribute) (SuperphotConfig attribute) NLIVE (in module src.superphot_plus.constants) normalization_means (ModelConfig attribute) (SuperphotConfig attribute) normalization_stddevs (ModelConfig attribute) (SuperphotConfig attribute)

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

O

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

oversample() (PosteriorSamplesGroup method) OVERSAMPLE_SIZE (in module src.superphot_plus.plotting.sampling_results) oversample_smote() (PosteriorSamplesGroup method)

P

pad_bands() (Lightcurve method) PAD_SIZE (in module src.superphot_plus.constants) param_labels() (in module src.superphot_plus.plotting.format_params) params_valid() (in module src.superphot_plus.utils) parsnip module plot_agreement_matrix_from_arrs() (in module src.superphot_plus.plotting.confusion_matrices) plot_binary_confusion_matrix() (in module src.superphot_plus.plotting.confusion_matrices) plot_calibration_curve() (in module src.superphot_plus.plotting.classifier_results) plot_chisquared_vs_accuracy() (in module src.superphot_plus.plotting.classifier_results) plot_class_fractions() (in module src.superphot_plus.plotting.classifier_results) plot_combined_posterior_space() (in module src.superphot_plus.plotting.sampling_results) plot_confusion_matrix() (in module src.superphot_plus.plotting.confusion_matrices) plot_corner_plot_all() (in module src.superphot_plus.plotting.sampling_results) plot_expected_agreement_matrix() (in module src.superphot_plus.plotting.confusion_matrices) plot_f1_curve() (in module src.superphot_plus.plotting.classifier_results) plot_feature_pacmap() (in module src.superphot_plus.plotting.sampling_results) plot_feature_umap() (in module src.superphot_plus.plotting.sampling_results) plot_high_confidence_confusion_matrix() (in module src.superphot_plus.plotting.confusion_matrices) plot_lc_fit() (in module src.superphot_plus.plotting.lightcurves) plot_lightcurve_clipping() (in module src.superphot_plus.plotting.lightcurves) plot_matrices() (in module src.superphot_plus.plotting.confusion_matrices) plot_metrics_over_mjd() (in module src.superphot_plus.plotting.classifier_results) plot_model_metrics() (in module src.superphot_plus.plotting.classifier_results)

plot_oversampling_1d() (in module src.superphot_plus.plotting.sampling_results) plot_param_distributions() (in module src.superphot_plus.plotting.sampling_results) plot_phase_vs_accuracy() (in module src.superphot_plus.plotting.classifier_results) plot_posterior_hist_numpyro_dict() (in module src.superphot_plus.plotting.sampling_results) plot_precision_recall() (in module src.superphot_plus.plotting.classifier_results) plot_redshifts_abs_mags() (in module src.superphot_plus.plotting.classifier_results) plot_sampling_lc_fit() (in module src.superphot_plus.plotting.lightcurves) plot_sampling_lc_fit_numpyro() (in module src.superphot_plus.plotting.lightcurves) plot_sampling_trace_numpyro() (in module src.superphot_plus.plotting.sampling_results) plot_snr_hist() (in module src.superphot_plus.plotting.classifier_results) plot_snr_npoints_vs_accuracy() (in module src.superphot_plus.plotting.classifier_results) plot_true_agreement_matrix() (in module src.superphot_plus.plotting.confusion_matrices) posterior_objects (PosteriorSamplesGroup attribute) PosteriorSamples (class in src.superphot_plus.posterior_samples) PosteriorSamplesGroup (class in src.superphot_plus.model.data) PosteriorsGenerator (class in scripts.generate_fits) prefix (SuperphotConfig attribute) prep_lcs_superraenn() (in module superraenn) print_last() (ModelMetrics method), [1] prior_helper() (in module src.superphot_plus.samplers.numpyro_sampler) PriorFields (class in src.superphot_plus.surveys.fitting_priors) priors (Survey attribute) prob_threshhold (SuperphotConfig attribute) probs_dir (SuperphotConfig attribute) probs_fn (SuperphotConfig attribute)

R

random_seed (PosteriorSamplesGroup attribute) read_probs_csv() (in module src.superphot_plus.plotting.utils) rebin_prec_recall() (in module src.superphot_plus.plotting.utils) redshifts (ZtfData attribute) ref_band_index (MultibandPriors property) reference_band (MultibandPriors attribute) relative_dirs (SuperphotConfig attribute) retrieve_decodings() (in module parsnip) (in module superraenn) retrieve_four_class_info() (in module src.superphot_plus.plotting.utils) retrieve_posterior_set() (in module src.superphot_plus.format_data_ztf) return_new_classifications() (SuperphotClassifier method) (SuperphotTrainer method) roc_curve_w_uncertainties() (in module src.superphot_plus.plotting.utils) run() (SuperphotTrainer method) (SuperphotTuner method)

run_cross_validation() (SuperphotTuner method) run_fit() (in module src.superphot_plus.samplers.iminuit_sampler) run_mcmc() (in module src.superphot_plus.samplers.dynesty_sampler) (in module src.superphot_plus.samplers.numpyro_sampler) run_multi_curve() (DynestySampler method) (IminuitSampler method) (LiCuSampler method) (NumpyroSampler method) (Sampler method) run_sampler() (PosteriorsGenerator method) run_single_curve() (DynestySampler method) (IminuitSampler method) (LiCuSampler method) (NumpyroSampler method) (Sampler method) run_superraenn_raenn() (in module superraenn)

S

sample_mean() (PosteriorSamples method) Sampler (class in src.superphot_plus.samplers.sampler) save() (SuperphotClassifier method) (SuperphotLightGBM method) (SuperphotMLP method) save_class_fractions() (in module src.superphot_plus.plotting.classifier_results) save_test_probabilities() (in module src.superphot_plus.utils) save_to_file() (Lightcurve method) (PosteriorSamples method) scripts module scripts.generate_fits module SEED (in module src.superphot_plus.constants) set_best_val_loss() (ModelConfig method) (SuperphotConfig method) set_global_plot_formatting() (in module src.superphot_plus.plotting.format_params) set_non_tunable_params() (ModelConfig method) (SuperphotConfig method) setup_model() (SuperphotTrainer method) setup_sampler() (PosteriorsGenerator method) SMALL_SIZE (in module src.superphot_plus.constants) sort_by_time() (Lightcurve method) split() (PosteriorSamplesGroup method) split_train_test() (TrainerBase method) src module src.superphot_plus module src.superphot_plus.config 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_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.data module src.superphot_plus.model.lightgbm module src.superphot_plus.model.metrics module src.superphot_plus.model.mlp module src.superphot_plus.plotting module src.superphot_plus.plotting.classifier_results module src.superphot_plus.plotting.confusion_matrices module src.superphot_plus.plotting.format_params module src.superphot_plus.plotting.lightcurves module

src.superphot_plus.plotting.sampling_results module src.superphot_plus.plotting.utils module src.superphot_plus.posterior_samples module src.superphot_plus.samplers module src.superphot_plus.samplers.dynesty_sampler module src.superphot_plus.samplers.iminuit_sampler module src.superphot_plus.samplers.licu_sampler module src.superphot_plus.samplers.numpyro_sampler module src.superphot_plus.samplers.sampler module src.superphot_plus.sfd module src.superphot_plus.sfd.model module src.superphot_plus.sfd.model.classifier module src.superphot_plus.sfd.model.config module src.superphot_plus.sfd.model.data module src.superphot_plus.sfd.model.metrics module src.superphot_plus.supernova_class module src.superphot_plus.surveys module src.superphot_plus.surveys.fitting_priors module src.superphot_plus.surveys.surveys 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) superphot module SuperphotClassifier (class in src.superphot_plus.sfd.model.classifier) SuperphotConfig (class in src.superphot_plus.config) SuperphotLightGBM (class in src.superphot_plus.model.lightgbm) SuperphotMLP (class in src.superphot_plus.model.mlp) SuperphotTrainer (class in src.superphot_plus.trainer) SuperphotTuner (class in src.superphot_plus.tuner) superraenn module suppress_stdout() (in module superphot) Survey (class in src.superphot_plus.surveys.surveys)

T

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

train_acc (ModelMetrics attribute), [1] train_and_validate() (SuperphotClassifier method) (SuperphotLightGBM method) (SuperphotMLP method) train_dataset (TrainData attribute), [1] train_epoch() (SuperphotClassifier method) (SuperphotMLP method) train_loss (ModelMetrics attribute), [1] train_parsnip_model() (in module parsnip) TrainData (class in src.superphot_plus.model.data) (class in src.superphot_plus.sfd.model.data) 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 src.superphot_plus.samplers.numpyro_sampler) tune_model() (SuperphotTuner method)

U

unique_bands() (Lightcurve method)

use_redshift_info (PosteriorSamplesGroup attribute)

V

val_acc (ModelMetrics attribute), [1] val_dataset (TrainData attribute), [1]

val_loss (ModelMetrics attribute), [1] VALID_RATIO (in module src.superphot_plus.constants) villar_fit_constraint() (in module src.superphot_plus.utils)

W

wavelengths (Survey attribute) write_metrics_to_file() (in module src.superphot_plus.utils) write_to_file() (ModelConfig method) (SuperphotConfig method) (Survey method)

wrongly_classified_dir (SuperphotConfig attribute)

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.sfd.model.data)