Index A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | R | S | T | U | V | W | Z A angstrom_to_micron() (in module specsr.models.lines) apply_env_overrides() (in module specsr.runtime) archive_dir() (in module specsr.checkpoints) archive_path() (in module specsr.checkpoints) as_dict() (specsr.training.ztransform.RedshiftTransform method) assert_clean_3way() (in module specsr.data.splits) assert_no_group_leakage() (in module specsr.data.splits) augment (specsr.data.build.BuildConfig attribute) augment_pair() (in module specsr.data.augment) augment_train_only (specsr.data.build.BuildConfig attribute) AugmentationConfig (class in specsr.data.augment) available_archives() (in module specsr.checkpoints) available_checkpoints() (in module specsr.checkpoints) B bounded_mean (specsr.models.zhead.ZHead1D property) build_dataset() (in module specsr.data.build) build_line_mask() (in module specsr.models.sr2) build_param_groups() (in module specsr.models.blocks) build_source() (in module specsr.training.zhead_sources) build_sr2_input() (in module specsr.models.sr2) BuildConfig (class in specsr.data.build) C cache_dir() (in module specsr.paths) centers() (specsr.data.grid.LogWavelengthGrid method) compute_psd_stats() (in module specsr.metrics) constrain_delta() (in module specsr.models.sr2) coverage (specsr.data.stitch.StitchedSpectrum property) D data_dir() (in module specsr.paths) decode_mean() (specsr.training.ztransform.RedshiftTransform method) decode_sigma() (specsr.training.ztransform.RedshiftTransform method) DEFAULT_GRID (in module specsr.data.grid) delta_cap (specsr.training.zhead_sources.SR2Source attribute) denormalize() (specsr.data.datasets.PairedSpectra method) (specsr.training.ztransform.RedshiftTransform method) describe_overrides() (in module specsr.runtime) discover_spectra() (in module specsr.data.ingest) disperser (specsr.data.ingest.SpectrumFile attribute) doublet_dip_depth() (in module specsr.metrics) E edges() (specsr.data.grid.LogWavelengthGrid method) ensure_archive() (in module specsr.checkpoints) env_int() (in module specsr.runtime) estimate_median_noise_lambda() (in module specsr.metrics) F field (specsr.data.ingest.SpectrumFile attribute) fields (specsr.data.build.BuildConfig attribute) finite_diff() (in module specsr.training.losses) finite_diff2() (in module specsr.training.losses) fit_line_sideband_weighted() (in module specsr.linefit) FixedGridSpectraDataset (class in specsr.data.datasets) flux (specsr.data.stitch.StitchedSpectrum attribute) flux_err (specsr.data.stitch.StitchedSpectrum attribute) forward() (specsr.models.blocks.ResidualBlock1D method) (specsr.models.sr1.SuperRes1D method) (specsr.models.sr2.SR2Attention method) (specsr.models.zhead.ZHead1D method) from_checkpoints() (specsr.inference.pipeline.SpecSRPipeline class method) from_dict() (specsr.training.ztransform.RedshiftTransform class method) from_pretrained() (specsr.inference.pipeline.SpecSRPipeline class method) from_training_redshifts() (specsr.training.ztransform.RedshiftTransform class method) G gather_line_windows() (in module specsr.training.losses) gauss_lin() (in module specsr.linefit) gaussian_line_mask() (in module specsr.models.sr2) get_activation() (in module specsr.models.blocks) get_checkpoint() (in module specsr.checkpoints) get_or_make_split() (in module specsr.data.splits) get_or_make_split_3way() (in module specsr.data.splits) get_training_split() (in module specsr.data.splits) grid (specsr.data.build.BuildConfig attribute) group_by_target() (in module specsr.data.ingest) H hash_file() (in module specsr.data.splits) heteroscedastic_nll() (in module specsr.models.zhead) highpass() (in module specsr.models.blocks) I is_prism (specsr.data.ingest.SpectrumFile property) is_smoke_run() (in module specsr.runtime) J jades_root() (in module specsr.paths) K keep_only_wide() (in module specsr.training.losses) key (specsr.data.ingest.SpectrumFile property) (specsr.training.zhead_sources.RawFluxSource attribute) L lambda_max (specsr.data.grid.LogWavelengthGrid attribute) lambda_min (specsr.data.grid.LogWavelengthGrid attribute) largest_divisor_at_most() (in module specsr.models.blocks) limit (specsr.data.build.BuildConfig attribute) limit_batches() (in module specsr.runtime) line_positions() (specsr.models.sr2.SR2Attention method) line_presence_target() (in module specsr.training.losses) line_snr_from_fit() (in module specsr.linefit) line_wavelengths_um() (in module specsr.models.lines) line_window_flux() (in module specsr.training.losses) load_config() (in module specsr.config) logits() (specsr.models.zhead.ZHead1D method) LogWavelengthGrid (class in specsr.data.grid) M mad_sigma() (in module specsr.linefit) make_group_split() (in module specsr.data.splits) make_group_split_3way() (in module specsr.data.splits) make_line_mask_from_smoothed() (in module specsr.training.losses) masked_mean() (in module specsr.training.losses) mean (specsr.training.ztransform.RedshiftTransform attribute) measure_line_snr() (in module specsr.linefit) merge_overrides() (in module specsr.config) min_coverage (specsr.data.build.BuildConfig attribute) min_valid_fraction (specsr.data.build.BuildConfig attribute) module specsr.checkpoints specsr.config specsr.data.augment specsr.data.build specsr.data.datasets specsr.data.grid specsr.data.ingest specsr.data.splits specsr.data.stitch specsr.inference.pipeline specsr.linefit specsr.metrics specsr.models.blocks specsr.models.lines specsr.models.sr1 specsr.models.sr2 specsr.models.zhead specsr.paths specsr.runtime specsr.training.losses specsr.training.zhead_sources specsr.training.ztransform moments() (specsr.models.zhead.ZHead1D method) moving_average_rows() (in module specsr.metrics) N n_arms (specsr.data.stitch.StitchedSpectrum attribute) n_aug (specsr.data.augment.AugmentationConfig attribute) n_channels (specsr.training.zhead_sources.RawFluxSource attribute) (specsr.training.zhead_sources.SR1Source property) (specsr.training.zhead_sources.SR2Source property) (specsr.training.zhead_sources.ZHeadSource attribute) n_samples (specsr.data.datasets.PairedSpectra property) (specsr.data.grid.LogWavelengthGrid property) name (specsr.training.zhead_sources.RawFluxSource attribute) (specsr.training.zhead_sources.SR1Source attribute) (specsr.training.zhead_sources.SR2Source attribute) (specsr.training.zhead_sources.ZHeadSource attribute) noise_frac (specsr.data.augment.AugmentationConfig attribute) normalize() (specsr.training.ztransform.RedshiftTransform method) normalize_spectrum() (in module specsr.data.datasets) O odd_kernel() (in module specsr.models.blocks) output_dir() (in module specsr.paths) P PairedSpectra (class in specsr.data.datasets) parent_group_ids() (in module specsr.data.splits) path (specsr.data.ingest.SpectrumFile attribute) predict() (specsr.training.ztransform.RedshiftTransform method) PRISM (in module specsr.data.ingest) R rank_doublet_examples() (in module specsr.metrics) RawFluxSource (class in specsr.training.zhead_sources) read_spectrum() (in module specsr.data.ingest) redshift_metrics() (in module specsr.metrics) redshift_pdf_loss() (in module specsr.models.zhead) RedshiftTransform (class in specsr.training.ztransform) release (specsr.data.build.BuildConfig attribute) release_dir() (in module specsr.paths) require_gratings (specsr.data.build.BuildConfig attribute) require_jades_root() (in module specsr.paths) REQUIRED (specsr.data.datasets.PairedSpectra attribute) resample_flux_conserving() (in module specsr.data.grid) resample_with_mask() (in module specsr.data.stitch) ResidualBlock1D (class in specsr.models.blocks) resolve_dataset() (in module specsr.runtime) resolve_zhead_in() (in module specsr.inference.pipeline) resolving_power (specsr.data.grid.LogWavelengthGrid attribute) robust_mad() (in module specsr.training.losses) robust_sigma_lambda() (in module specsr.metrics) robust_symmetric_vlim() (in module specsr.metrics) run_build() (in module specsr.data.build) run_infer() (in module specsr.inference.pipeline) S samples_per_resolution_element() (specsr.data.grid.LogWavelengthGrid method) sampling_report() (in module specsr.data.grid) secure_z_flags (specsr.data.build.BuildConfig attribute) seed (specsr.data.augment.AugmentationConfig attribute) shift_redshift_on_log_grid() (in module specsr.data.augment) sigma_base_um (specsr.training.zhead_sources.SR2Source attribute) sigma_z (specsr.data.augment.AugmentationConfig attribute) smooth1d() (in module specsr.models.blocks) specsr.checkpoints module specsr.config module specsr.data.augment module specsr.data.build module specsr.data.datasets module specsr.data.grid module specsr.data.ingest module specsr.data.splits module specsr.data.stitch module specsr.inference.pipeline module specsr.linefit module specsr.metrics module specsr.models.blocks module specsr.models.lines module specsr.models.sr1 module specsr.models.sr2 module specsr.models.zhead module specsr.paths module specsr.runtime module specsr.training.losses module specsr.training.zhead_sources module specsr.training.ztransform module SpecSRPipeline (class in specsr.inference.pipeline) SpecSRResult (class in specsr.inference.pipeline) SpectrumFile (class in specsr.data.ingest) SPLIT_SCHEME_3WAY (in module specsr.data.splits) split_seed (specsr.data.build.BuildConfig attribute) sr1 (specsr.inference.pipeline.SpecSRResult attribute) (specsr.training.zhead_sources.SR1Source attribute) (specsr.training.zhead_sources.SR2Source attribute) sr1_deblend_loss() (in module specsr.training.losses) sr1_sigma (specsr.inference.pipeline.SpecSRResult attribute) SR1Source (class in specsr.training.zhead_sources) sr2 (specsr.inference.pipeline.SpecSRResult attribute) (specsr.training.zhead_sources.SR2Source attribute) sr2_input_channels() (in module specsr.models.sr2) sr2_loss() (in module specsr.training.losses) sr2_sigma (specsr.inference.pipeline.SpecSRResult attribute) SR2Attention (class in specsr.models.sr2) SR2Source (class in specsr.training.zhead_sources) std (specsr.training.ztransform.RedshiftTransform attribute) stitch_gratings() (in module specsr.data.stitch) StitchedSpectrum (class in specsr.data.stitch) SuperRes1D (class in specsr.models.sr1) T target_id (specsr.data.ingest.SpectrumFile attribute) tier (specsr.data.ingest.SpectrumFile attribute) train_frac (specsr.data.build.BuildConfig attribute) try_archive() (in module specsr.checkpoints) U unwrap_wandb_config() (in module specsr.config) use_line_mask (specsr.training.zhead_sources.SR2Source attribute) use_sigma (specsr.training.zhead_sources.SR1Source attribute) (specsr.training.zhead_sources.SR2Source attribute) use_sr1_sigma (specsr.training.zhead_sources.SR2Source attribute) use_zhat_channel (specsr.training.zhead_sources.SR2Source attribute) use_zsigma_channel (specsr.training.zhead_sources.SR2Source attribute) V valid (specsr.data.stitch.StitchedSpectrum attribute) W wandb_mode() (in module specsr.runtime) wavelength (specsr.data.stitch.StitchedSpectrum attribute) (specsr.inference.pipeline.SpecSRPipeline property) (specsr.inference.pipeline.SpecSRResult attribute) Z z (specsr.inference.pipeline.SpecSRResult attribute) z_max_n (specsr.training.ztransform.RedshiftTransform attribute) z_min_n (specsr.training.ztransform.RedshiftTransform attribute) z_sigma (specsr.inference.pipeline.SpecSRResult attribute) ZHead1D (class in specsr.models.zhead) zhead_bootstrap (specsr.training.zhead_sources.SR2Source attribute) ZHeadSource (class in specsr.training.zhead_sources) zsigma_line_mask (specsr.training.zhead_sources.SR2Source attribute) zsigma_mask_max_um (specsr.training.zhead_sources.SR2Source attribute) ztransform (specsr.training.zhead_sources.SR2Source attribute)