Index

A

ab_h158() (in module specsr_roman.extraction.catalog) AblationConfig (class in specsr_roman.evaluation.ablation)

add_grism_noise() (in module specsr_roman.extraction.simulate) angstrom_to_micron() (in module specsr_roman.lines) apply_phot_noise() (in module specsr_roman.data.photometry)

B

band_names() (in module specsr_roman.data.photometry) build_line_mask() (in module specsr_roman.models.sr2)

build_prediction_cache() (in module specsr_roman.evaluation.cache) build_sr2_input() (in module specsr_roman.inference.pipeline)

C

cache_dir() (in module specsr_roman.paths) CacheConfig (class in specsr_roman.evaluation.cache) calzetti_k() (in module specsr_roman.data.augment)

CANONICAL_CHAIN (in module specsr_roman.checkpoints) constrain_delta() (in module specsr_roman.models.sr2) count_strong_lines() (in module specsr_roman.lines)

D

data_root() (in module specsr_roman.paths) dataset_dir() (in module specsr_roman.paths)

default_split_dir() (in module specsr_roman.data.splits) detect_and_relabel() (in module specsr_roman.extraction.catalog) disperse_scene() (in module specsr_roman.extraction.simulate)

E

eval_mag_err (specsr_roman.evaluation.cache.CacheConfig attribute) extract_target() (in module specsr_roman.extraction.extract)

ExtractionConfig (class in specsr_roman.extraction.batch) ExtractionFailure

F

filter_split_min_lines() (in module specsr_roman.data.splits) find_line_segments() (in module specsr_roman.data.augment) fluxconserve_resample() (in module specsr_roman.data.transforms)

forward() (specsr_roman.models.sr2.SR2Attention method) from_pretrained() (specsr_roman.inference.pipeline.RomanPipeline class method) from_state_dict() (specsr_roman.models.sr1.SuperRes1D class method) (specsr_roman.models.zhead.ZHeadClf class method)

G

get_or_make_group_split() (in module specsr_roman.data.splits) get_or_make_split() (in module specsr_roman.data.splits)

grizli_conf_dir() (in module specsr_roman.paths) grizli_spectrum() (specsr_roman.extraction.seds.SEDLibrary method)

H

hash_file() (in module specsr_roman.data.splits)

I

interp_ascending() (in module specsr_roman.data.transforms)

L

line_amplitude_recovery() (in module specsr_roman.evaluation.metrics) line_flux_loss() (in module specsr_roman.training.losses) line_flux_loss_weighted() (in module specsr_roman.training.losses) line_hallucination_loss() (in module specsr_roman.training.losses) line_profiles() (in module specsr_roman.models.sr2) line_snr_from_spectrum() (in module specsr_roman.evaluation.metrics)

load_config() (in module specsr_roman.config) load_sr1() (in module specsr_roman.checkpoints) load_sr2() (in module specsr_roman.checkpoints) load_truth_index() (in module specsr_roman.extraction.catalog) load_zhead() (in module specsr_roman.models.zhead) load_zhead_ckpt() (in module specsr_roman.checkpoints) loss_deblend_gated() (in module specsr_roman.training.losses)

M

make_line_mask_from_smoothed() (in module specsr_roman.training.losses) make_z_grid() (in module specsr_roman.models.zhead) merge() (in module specsr_roman.extraction.batch) module specsr_roman.checkpoints specsr_roman.config specsr_roman.data.augment specsr_roman.data.datasets specsr_roman.data.photometry specsr_roman.data.splits specsr_roman.data.transforms specsr_roman.evaluation.ablation specsr_roman.evaluation.cache specsr_roman.evaluation.metrics specsr_roman.evaluation.prior_dominance specsr_roman.extraction.batch specsr_roman.extraction.catalog specsr_roman.extraction.extract specsr_roman.extraction.frames specsr_roman.extraction.seds specsr_roman.extraction.simulate specsr_roman.grids specsr_roman.inference.pipeline specsr_roman.lines specsr_roman.models.sr1 specsr_roman.models.sr2 specsr_roman.models.zhead specsr_roman.paths specsr_roman.training.losses specsr_roman.training.sr1 specsr_roman.training.sr2 specsr_roman.training.zhead

N

normalize() (in module specsr_roman.data.transforms)

O

observed() (specsr_roman.extraction.seds.SEDLibrary method)

P

per_line_amplitude_recovery() (in module specsr_roman.evaluation.metrics) PipelineResult (class in specsr_roman.inference.pipeline) plot_ablation() (in module specsr_roman.evaluation.ablation) predict() (specsr_roman.inference.pipeline.RomanPipeline method) predict_z() (specsr_roman.models.zhead.ZHeadClf method)

prepare_frames() (in module specsr_roman.extraction.frames) presence_labels() (in module specsr_roman.training.losses) PriorDominanceConfig (class in specsr_roman.evaluation.prior_dominance) push_checkpoint() (in module specsr_roman.checkpoints) pz_stats() (in module specsr_roman.models.zhead)

R

RECOVERABILITY_BINS (in module specsr_roman.evaluation.metrics) redshift_summary() (in module specsr_roman.evaluation.metrics) resolve_checkpoint() (in module specsr_roman.checkpoints) resolve_phot_tier() (in module specsr_roman.grids) RomanFixedGridDataset (class in specsr_roman.data.datasets)

RomanPipeline (class in specsr_roman.inference.pipeline) run_ablation() (in module specsr_roman.evaluation.ablation) run_batch() (in module specsr_roman.extraction.batch) run_prior_dominance() (in module specsr_roman.evaluation.prior_dominance) run_worker() (in module specsr_roman.extraction.batch) runs_dir() (in module specsr_roman.paths)

S

SEDLibrary (class in specsr_roman.extraction.seds) select_bands() (in module specsr_roman.data.photometry) smooth_to_grism() (in module specsr_roman.data.transforms) soft_labels() (in module specsr_roman.models.zhead) specsr_roman.checkpoints module specsr_roman.config module specsr_roman.data.augment module specsr_roman.data.datasets module specsr_roman.data.photometry module specsr_roman.data.splits module specsr_roman.data.transforms module specsr_roman.evaluation.ablation module specsr_roman.evaluation.cache module specsr_roman.evaluation.metrics module specsr_roman.evaluation.prior_dominance module specsr_roman.extraction.batch module specsr_roman.extraction.catalog module specsr_roman.extraction.extract module specsr_roman.extraction.frames module

specsr_roman.extraction.seds module specsr_roman.extraction.simulate module specsr_roman.grids module specsr_roman.inference.pipeline module specsr_roman.lines module specsr_roman.models.sr1 module specsr_roman.models.sr2 module specsr_roman.models.zhead module specsr_roman.paths module specsr_roman.training.losses module specsr_roman.training.sr1 module specsr_roman.training.sr2 module specsr_roman.training.zhead module SpectrumAugmentor (class in specsr_roman.data.augment) SR1Config (class in specsr_roman.config) sr2_reconstruction_loss() (in module specsr_roman.training.losses) SR2Attention (class in specsr_roman.models.sr2) SR2Config (class in specsr_roman.config) standardization_stats() (in module specsr_roman.data.photometry) SuperRes1D (class in specsr_roman.models.sr1)

T

to_dict() (in module specsr_roman.config) topk_modes() (in module specsr_roman.models.sr2)

train() (in module specsr_roman.training.sr1) (in module specsr_roman.training.sr2) (in module specsr_roman.training.zhead)

Z

z_metrics() (in module specsr_roman.models.zhead) ZHead1D (class in specsr_roman.models.zhead) zhead_ckpt (specsr_roman.evaluation.ablation.AblationConfig attribute)

ZHeadAttn (class in specsr_roman.models.zhead) ZHeadClf (class in specsr_roman.models.zhead) ZHeadConfig (class in specsr_roman.config)