Index

A

AblationSaliencyAnalyzer (class in dorothy.analysis.saliency) AblationSaliencyResult (class in dorothy.analysis.saliency) accept_singular_keys() (dorothy.config.schema.DataConfig class method) ActivationType (class in dorothy.config.schema) AdditiveOutput (class in dorothy.models.output)

anomaly_fraction (dorothy.analysis.knn_anomaly.AnomalyResult property) AnomalyDetector (class in dorothy.analysis.knn_anomaly) AnomalyResult (class in dorothy.analysis.knn_anomaly) apply_quality_filter() (in module dorothy.data.fits_loader) autocast_dtype (dorothy.training.trainer.Trainer attribute)

B

Batch (class in dorothy.training.batches) batch_size (dorothy.analysis.saliency.AblationSaliencyAnalyzer attribute) batch_source() (dorothy.data.training_data.TrainingData method) batches() (dorothy.training.batches.SparseBatchSource method) (dorothy.training.batches.TensorBatchSource method) best_epoch (dorothy.training.trainer.TrainingHistory attribute)

best_val_loss (dorothy.training.trainer.TrainingHistory attribute) bias (dorothy.inference.evaluator.ParameterMetrics attribute) block_size (dorothy.analysis.saliency.AblationSaliencyResult attribute) build_batch() (dorothy.training.batches.SparseBatchSource method) build_batch_from_sparse() (in module dorothy.inference.evaluation_utils) build_model() (in module dorothy.inference.checkpoint) by_survey (dorothy.inference.evaluator.SurveyEvaluationResult attribute)

C

calibrate_scatter() (in module dorothy.inference.calibration) calibrated (dorothy.inference.predictor.PredictionResult attribute) calibration (dorothy.inference.checkpoint.LoadedCheckpoint attribute) (dorothy.inference.evaluation_utils.HeadEvaluation attribute) (dorothy.inference.predictor.Predictor attribute) (dorothy.training.trainer.Trainer attribute) calibration factor CatalogueData (class in dorothy.data.catalogue_loader) CatalogueInfo (class in dorothy.data.catalogue_loader) CatalogueLoader (class in dorothy.data.catalogue_loader) checkpoint combination_mode (dorothy.models.multi_head_mlp.MultiHeadMLP attribute) CombinationMode (class in dorothy.config.schema)

compare_label_sources() (in module dorothy.analysis.label_comparison) complete_config_from_catalogue() (in module dorothy.data.training_data) compute_saliency() (dorothy.analysis.saliency.AblationSaliencyAnalyzer method) (dorothy.analysis.saliency.SaliencyAnalyzer method) config (dorothy.inference.checkpoint.LoadedCheckpoint attribute) (dorothy.inference.predictor.Predictor attribute) (dorothy.training.trainer.Trainer attribute) configure_model() (dorothy.data.training_data.TrainingData method) count_parameters() (dorothy.models.mlp.MLP method) (dorothy.models.multi_head_mlp.MultiHeadMLP method) count_parameters_by_component() (dorothy.models.multi_head_mlp.MultiHeadMLP method) coverage() (dorothy.data.catalogue_loader.SparseMergedData method) creation_date (dorothy.data.catalogue_loader.CatalogueInfo attribute)

D

DataConfig (class in dorothy.config.schema) DataSplit (class in dorothy.data.training_data) dec (dorothy.analysis.saliency.AblationSaliencyResult attribute) (dorothy.analysis.saliency.SaliencyResult attribute) (dorothy.data.catalogue_loader.CatalogueData attribute) (dorothy.data.catalogue_loader.MergedCatalogueData attribute) (dorothy.data.catalogue_loader.SparseMergedData attribute) default_parameter_names() (in module dorothy.config.schema) delta_mu (dorothy.analysis.saliency.AblationSaliencyResult attribute) delta_mu_total (dorothy.analysis.saliency.AblationSaliencyResult attribute) delta_sigma (dorothy.analysis.saliency.AblationSaliencyResult attribute) delta_sigma_total (dorothy.analysis.saliency.AblationSaliencyResult attribute) detect() (dorothy.analysis.knn_anomaly.AnomalyDetector method) device (dorothy.analysis.saliency.AblationSaliencyAnalyzer attribute) (dorothy.analysis.saliency.SaliencyAnalyzer attribute) (dorothy.inference.checkpoint.LoadedCheckpoint attribute) (dorothy.inference.predictor.Predictor attribute) (dorothy.training.trainer.Trainer attribute) distances (dorothy.analysis.knn_anomaly.AnomalyResult attribute) dorothy module dorothy.analysis module dorothy.analysis.knn_anomaly module dorothy.analysis.label_comparison module dorothy.analysis.saliency module dorothy.cli module dorothy.config module dorothy.config.schema module dorothy.data module dorothy.data.augmentation module dorothy.data.catalogue_loader module dorothy.data.fits_loader module dorothy.data.normalizer module

dorothy.data.training_data module dorothy.inference module dorothy.inference.calibration module dorothy.inference.checkpoint module dorothy.inference.evaluation_utils module dorothy.inference.evaluator module dorothy.inference.predictor module dorothy.losses module dorothy.losses.heteroscedastic module dorothy.models module dorothy.models.mlp module dorothy.models.multi_head_mlp module dorothy.models.output module dorothy.models.utils module dorothy.training module dorothy.training.batches module dorothy.training.trainer module dorothy.utils module dorothy.visualization module dorothy.visualization.evaluation_plots module dorothy.visualization.training_plots module DynamicInputMasking (class in dorothy.data.augmentation) DynamicLabelMasking (class in dorothy.data.augmentation)

E

element_losses() (dorothy.losses.heteroscedastic.HeteroscedasticLoss method) (dorothy.losses.heteroscedastic.MaskedMSELoss method) encode_precision() (in module dorothy.data.fits_loader) encode_present_rows_only (dorothy.models.multi_head_mlp.MultiHeadMLP attribute) encoder encoders (dorothy.models.multi_head_mlp.MultiHeadMLP attribute) error_projection (dorothy.models.output.LabelAwareOutput attribute) errors (dorothy.data.fits_loader.SpectralData attribute) evaluate() (dorothy.inference.evaluator.Evaluator method) evaluate_by_survey() (dorothy.inference.evaluator.Evaluator method) evaluate_model() (in module dorothy.inference.evaluation_utils)

evaluate_on_dense_data() (in module dorothy.inference.evaluation_utils) evaluate_on_single_survey_data() (in module dorothy.inference.evaluation_utils) evaluate_on_test_set() (in module dorothy.inference.evaluation_utils) evaluate_predictions() (in module dorothy.inference.evaluator) evaluate_predictions_by_survey() (in module dorothy.inference.evaluator) EvaluationResult (class in dorothy.inference.evaluator) Evaluator (class in dorothy.inference.evaluator) ExperimentConfig (class in dorothy.config.schema) extra_repr() (dorothy.losses.heteroscedastic.HeteroscedasticLoss method) (dorothy.models.mlp.MLP method) (dorothy.models.multi_head_mlp.MultiHeadMLP method)

F

f_max (dorothy.data.augmentation.DynamicInputMasking attribute) f_min_override (dorothy.data.augmentation.DynamicInputMasking attribute) filter_by_flags() (dorothy.data.catalogue_loader.CatalogueData method) filter_by_surveys() (dorothy.data.catalogue_loader.MergedCatalogueData method) filter_complete() (dorothy.data.catalogue_loader.CatalogueData method) fisher_importance (dorothy.analysis.saliency.SaliencyResult attribute) fisher_importance_per_param (dorothy.analysis.saliency.SaliencyResult attribute) fisher_weighted (dorothy.analysis.saliency.AblationSaliencyResult attribute) fisher_weighted_total (dorothy.analysis.saliency.AblationSaliencyResult attribute) fit() (dorothy.analysis.knn_anomaly.AnomalyDetector method) (dorothy.data.normalizer.LabelNormalizer method) (dorothy.data.training_data.TrainingData method) (dorothy.training.trainer.Trainer method) fit_multi_labelset() (dorothy.training.trainer.Trainer method) fit_multi_survey() (dorothy.training.trainer.Trainer method) fit_multi_survey_sparse() (dorothy.training.trainer.Trainer method) fits_path (dorothy.data.fits_loader.FITSLoader attribute) FITSLoader (class in dorothy.data.fits_loader) flag flux (dorothy.data.catalogue_loader.CatalogueData attribute) (dorothy.data.catalogue_loader.SparseMergedData attribute) (dorothy.data.fits_loader.SpectralData attribute)

forward() (dorothy.losses.heteroscedastic.HeteroscedasticLoss method) (dorothy.losses.heteroscedastic.MaskedMSELoss method) (dorothy.models.mlp.MLP method) (dorothy.models.multi_head_mlp.MultiHeadMLP method) (dorothy.models.multi_head_mlp.OutputHead method) (dorothy.models.multi_head_mlp.SharedTrunk method) (dorothy.models.multi_head_mlp.SurveyEncoder method) (dorothy.models.output.AdditiveOutput method) (dorothy.models.output.LabelAwareOutput method) (dorothy.models.utils.ResidualBlock method) forward_all_label_sources() (dorothy.models.multi_head_mlp.MultiHeadMLP method) forward_detailed() (dorothy.losses.heteroscedastic.HeteroscedasticLoss method) (dorothy.losses.heteroscedastic.MaskedMSELoss method) forward_for_label_source() (dorothy.models.multi_head_mlp.MultiHeadMLP method) forward_label_sources() (in module dorothy.training.batches) forward_multi() (dorothy.models.multi_head_mlp.MultiHeadMLP method) forward_single() (dorothy.models.multi_head_mlp.MultiHeadMLP method) from_config() (dorothy.inference.predictor.Predictor class method) (dorothy.models.mlp.MLP class method) (dorothy.models.multi_head_mlp.MultiHeadMLP class method) from_predictor() (dorothy.analysis.knn_anomaly.AnomalyDetector class method)

G

gaia_id (dorothy.analysis.saliency.AblationSaliencyResult attribute) (dorothy.analysis.saliency.SaliencyResult attribute) gaia_ids (dorothy.data.catalogue_loader.CatalogueData attribute) (dorothy.data.catalogue_loader.MergedCatalogueData attribute) (dorothy.data.catalogue_loader.SparseMergedData attribute) generate_evaluation_plots() (in module dorothy.visualization.evaluation_plots) generate_training_report() (in module dorothy.visualization.training_plots) get_activation() (in module dorothy.models.utils) get_anomaly_indices() (dorothy.analysis.knn_anomaly.AnomalyResult method) get_checkpoint_path() (dorothy.config.schema.ExperimentConfig method) get_coverage_stats() (dorothy.data.catalogue_loader.MergedCatalogueData method) (dorothy.data.catalogue_loader.SparseMergedData method) get_embeddings() (dorothy.inference.predictor.Predictor method) (dorothy.models.mlp.MLP method) (dorothy.models.multi_head_mlp.MultiHeadMLP method) get_errors() (dorothy.data.fits_loader.SpectralData method) get_grokking_metrics() (dorothy.training.trainer.TrainingHistory method) get_info() (dorothy.data.catalogue_loader.CatalogueLoader method) get_label_source_for_survey() (dorothy.data.catalogue_loader.CatalogueLoader method)

get_labels() (dorothy.data.fits_loader.SpectralData method) get_labels_for_source() (dorothy.data.catalogue_loader.SparseMergedData method) get_loss_breakdown_array() (dorothy.training.trainer.TrainingHistory method) get_metrics_array() (dorothy.training.trainer.TrainingHistory method) get_model_input_3channel() (dorothy.data.fits_loader.SpectralData method) get_normalization() (in module dorothy.models.utils) get_output_path() (dorothy.config.schema.ExperimentConfig method) get_overlap_matrix() (dorothy.data.catalogue_loader.MergedCatalogueData method) get_param_index() (dorothy.analysis.saliency.AblationSaliencyResult method) (dorothy.analysis.saliency.SaliencyResult method) get_parameter_names() (dorothy.data.catalogue_loader.CatalogueLoader method) get_params_dict() (dorothy.data.normalizer.LabelNormalizer method) get_predicted_scatter() (dorothy.losses.heteroscedastic.HeteroscedasticLoss method) get_predictions_and_uncertainties() (dorothy.training.trainer.Trainer method) get_survey_wavelength_counts() (dorothy.data.catalogue_loader.CatalogueLoader method) get_survey_wavelengths() (dorothy.data.catalogue_loader.CatalogueLoader method) get_training_distances() (dorothy.analysis.knn_anomaly.AnomalyDetector method) global_to_local (dorothy.data.catalogue_loader.SparseMergedData attribute) grad_norms (dorothy.training.trainer.TrainingHistory attribute)

H

has_data has_data() (dorothy.data.catalogue_loader.SparseMergedData method) has_labels (dorothy.data.catalogue_loader.CatalogueData attribute) has_labels_dict (dorothy.data.catalogue_loader.SparseMergedData attribute) has_labels_for_source() (dorothy.data.catalogue_loader.SparseMergedData method) has_spectrum (dorothy.data.catalogue_loader.CatalogueData attribute)

head head_label_sources() (in module dorothy.inference.checkpoint) HeadEvaluation (class in dorothy.inference.evaluation_utils) HeteroscedasticLoss (class in dorothy.losses.heteroscedastic) hidden_blocks() (in module dorothy.models.utils) hidden_layers (dorothy.models.mlp.MLP attribute) history (dorothy.training.trainer.Trainer attribute)

I

ids (dorothy.data.fits_loader.SpectralData attribute) infer_mlp_architecture() (in module dorothy.inference.checkpoint) info() (dorothy.data.catalogue_loader.CatalogueLoader method) input_dim (dorothy.models.multi_head_mlp.SharedTrunk attribute) input_features (dorothy.models.mlp.MLP attribute) InputMaskingConfig (class in dorothy.config.schema) inverse_transform() (dorothy.data.normalizer.LabelNormalizer method) iqr (dorothy.data.normalizer.ParameterStats attribute) is_anomaly (dorothy.analysis.knn_anomaly.AnomalyResult attribute) is_fitted (dorothy.data.normalizer.LabelNormalizer attribute)

is_multi_head (dorothy.config.schema.ExperimentConfig property) (dorothy.inference.predictor.Predictor property) is_multi_label (dorothy.config.schema.DataConfig property) (dorothy.config.schema.MultiHeadModelConfig property) is_multi_label() (dorothy.data.catalogue_loader.SparseMergedData method) is_multi_survey (dorothy.config.schema.DataConfig property) is_normalized (dorothy.inference.predictor.PredictionResult attribute) ivar (dorothy.data.catalogue_loader.CatalogueData attribute) (dorothy.data.fits_loader.SpectralData attribute) ivar_to_error_and_mask() (in module dorothy.data.fits_loader)

J

jacobian_mean_error (dorothy.analysis.saliency.SaliencyResult attribute) jacobian_mean_flux (dorothy.analysis.saliency.SaliencyResult attribute)

jacobian_sigma_error (dorothy.analysis.saliency.SaliencyResult attribute) jacobian_sigma_flux (dorothy.analysis.saliency.SaliencyResult attribute)

K

k (dorothy.analysis.knn_anomaly.AnomalyDetector attribute) (dorothy.analysis.knn_anomaly.AnomalyResult attribute)

kind (dorothy.data.training_data.TrainingData attribute)

L

l2_normalize() (in module dorothy.analysis.knn_anomaly) label label source label_aware_uncertainty (dorothy.models.mlp.MLP attribute) (dorothy.models.multi_head_mlp.MultiHeadMLP attribute) label_errors (dorothy.data.catalogue_loader.CatalogueData attribute) label_flags (dorothy.data.catalogue_loader.CatalogueData attribute) label_keys (dorothy.training.batches.SparseBatchSource property) (dorothy.training.batches.TensorBatchSource property) (dorothy.training.trainer.Trainer property) label_source (dorothy.data.catalogue_loader.CatalogueData attribute) (dorothy.data.catalogue_loader.MergedCatalogueData attribute) (dorothy.inference.evaluation_utils.HeadEvaluation attribute) (dorothy.inference.predictor.Predictor attribute) label_sources (dorothy.data.training_data.TrainingData attribute) (dorothy.models.multi_head_mlp.MultiHeadMLP attribute) LabelAwareOutput (class in dorothy.models.output) LabelComparison (class in dorothy.analysis.label_comparison) LabelMaskingConfig (class in dorothy.config.schema) LabelNormalizer (class in dorothy.data.normalizer) labels (dorothy.data.catalogue_loader.CatalogueData attribute) (dorothy.data.catalogue_loader.CatalogueInfo attribute) (dorothy.data.catalogue_loader.SparseMergedData attribute) (dorothy.data.fits_loader.SpectralData attribute) labels() (dorothy.data.training_data.TrainingData method) labels_dict (dorothy.data.catalogue_loader.SparseMergedData attribute) latent_dim (dorothy.models.multi_head_mlp.MultiHeadMLP attribute) (dorothy.models.multi_head_mlp.SurveyEncoder attribute) layer_index (dorothy.analysis.knn_anomaly.AnomalyDetector attribute) layers (dorothy.models.mlp.MLP attribute)

learning_rates (dorothy.training.trainer.TrainingHistory attribute) legacy_filtered_rows() (dorothy.data.training_data.TrainingData method) list_label_sources() (dorothy.data.catalogue_loader.CatalogueLoader method) list_surveys() (dorothy.data.catalogue_loader.CatalogueLoader method) load() (dorothy.data.catalogue_loader.CatalogueLoader method) (dorothy.data.fits_loader.FITSLoader method) (dorothy.data.normalizer.LabelNormalizer class method) (dorothy.data.training_data.DataSplit class method) (dorothy.inference.predictor.Predictor class method) load_calibration() (in module dorothy.inference.calibration) load_checkpoint() (in module dorothy.inference.checkpoint) load_checkpoint_config() (dorothy.config.schema.ExperimentConfig class method) load_for_training() (dorothy.data.catalogue_loader.CatalogueLoader method) load_labels() (dorothy.data.catalogue_loader.CatalogueLoader method) load_labels_for_index() (dorothy.data.catalogue_loader.CatalogueLoader method) load_merged() (dorothy.data.catalogue_loader.CatalogueLoader method) load_merged_for_training() (dorothy.data.catalogue_loader.CatalogueLoader method) load_merged_sparse() (dorothy.data.catalogue_loader.CatalogueLoader method) load_multi() (dorothy.data.catalogue_loader.CatalogueLoader method) load_raw_spectra() (dorothy.data.catalogue_loader.CatalogueLoader method) load_spectrum() (dorothy.data.catalogue_loader.CatalogueLoader method) load_state() (dorothy.analysis.knn_anomaly.AnomalyDetector method) load_training_data() (in module dorothy.data.training_data) load_yaml() (dorothy.config.schema.ExperimentConfig class method) LoadedCheckpoint (class in dorothy.inference.checkpoint) local_to_global (dorothy.data.catalogue_loader.SparseMergedData attribute) log_iqr (dorothy.data.normalizer.ParameterStats attribute) log_median (dorothy.data.normalizer.ParameterStats attribute) loss_fn (dorothy.training.trainer.Trainer attribute) LossType (class in dorothy.config.schema)

M

mae (dorothy.inference.evaluator.ParameterMetrics attribute) main() (in module dorothy.cli) mask (dorothy.analysis.saliency.AblationSaliencyResult attribute) (dorothy.analysis.saliency.SaliencyResult attribute) MaskedMSELoss (class in dorothy.losses.heteroscedastic) mean (dorothy.models.output.LabelAwareOutput attribute) median (dorothy.data.normalizer.ParameterStats attribute) median_offset (dorothy.analysis.label_comparison.LabelComparison attribute) (dorothy.inference.evaluator.ParameterMetrics attribute) memory_usage_mb() (dorothy.data.catalogue_loader.SparseMergedData method) MergedCatalogueData (class in dorothy.data.catalogue_loader) metrics (dorothy.inference.evaluator.EvaluationResult attribute) missing vector missing_log_error (dorothy.models.output.LabelAwareOutput attribute) MLP (class in dorothy.models.mlp) model (dorothy.analysis.saliency.AblationSaliencyAnalyzer attribute) (dorothy.analysis.saliency.SaliencyAnalyzer attribute) (dorothy.inference.checkpoint.LoadedCheckpoint attribute) (dorothy.inference.predictor.Predictor attribute) (dorothy.training.trainer.Trainer attribute) model_config (dorothy.config.schema.DataConfig attribute) (dorothy.config.schema.ExperimentConfig attribute) (dorothy.config.schema.InputMaskingConfig attribute) (dorothy.config.schema.LabelMaskingConfig attribute) (dorothy.config.schema.ModelConfig attribute) (dorothy.config.schema.MultiHeadModelConfig attribute) (dorothy.config.schema.OptimizerConfig attribute) (dorothy.config.schema.SchedulerConfig attribute) (dorothy.config.schema.TrainingConfig attribute) ModelConfig (class in dorothy.config.schema) module dorothy dorothy.analysis dorothy.analysis.knn_anomaly dorothy.analysis.label_comparison dorothy.analysis.saliency dorothy.cli dorothy.config dorothy.config.schema dorothy.data dorothy.data.augmentation dorothy.data.catalogue_loader dorothy.data.fits_loader dorothy.data.normalizer dorothy.data.training_data dorothy.inference dorothy.inference.calibration dorothy.inference.checkpoint dorothy.inference.evaluation_utils dorothy.inference.evaluator dorothy.inference.predictor dorothy.losses dorothy.losses.heteroscedastic dorothy.models dorothy.models.mlp dorothy.models.multi_head_mlp dorothy.models.output dorothy.models.utils dorothy.training dorothy.training.batches dorothy.training.trainer dorothy.utils dorothy.visualization dorothy.visualization.evaluation_plots dorothy.visualization.training_plots

MultiHeadMLP (class in dorothy.models.multi_head_mlp) MultiHeadModelConfig (class in dorothy.config.schema)

N

n_anomalies (dorothy.analysis.knn_anomaly.AnomalyResult property) n_complete (dorothy.data.catalogue_loader.CatalogueData property) n_parameters (dorothy.config.schema.ModelConfig property) (dorothy.data.fits_loader.SpectralData property) (dorothy.data.normalizer.LabelNormalizer property) (dorothy.inference.predictor.PredictionResult property) (dorothy.losses.heteroscedastic.HeteroscedasticLoss attribute) (dorothy.models.mlp.MLP attribute) (dorothy.models.multi_head_mlp.MultiHeadMLP attribute) (dorothy.models.multi_head_mlp.OutputHead attribute) (dorothy.training.trainer.Trainer property) n_params (dorothy.data.catalogue_loader.SparseMergedData attribute) n_rows (dorothy.data.training_data.TrainingData property) n_samples (dorothy.analysis.knn_anomaly.AnomalyResult property) (dorothy.data.fits_loader.SpectralData property) (dorothy.inference.evaluator.ParameterMetrics attribute) (dorothy.inference.predictor.PredictionResult property) n_shared (dorothy.analysis.label_comparison.LabelComparison attribute) n_stars (dorothy.data.catalogue_loader.CatalogueData property) (dorothy.data.catalogue_loader.CatalogueInfo attribute) (dorothy.data.catalogue_loader.MergedCatalogueData property)

n_total (dorothy.data.catalogue_loader.SparseMergedData attribute) n_wavelengths (dorothy.data.fits_loader.SpectralData property) (dorothy.models.multi_head_mlp.SurveyEncoder attribute) n_with_data() (dorothy.data.catalogue_loader.SparseMergedData method) n_with_labels (dorothy.data.catalogue_loader.CatalogueData property) n_with_spectra (dorothy.data.catalogue_loader.CatalogueData property) name (dorothy.data.normalizer.ParameterStats attribute) (dorothy.inference.evaluator.ParameterMetrics attribute) nll_warmup_epochs (dorothy.training.trainer.TrainingHistory attribute) NormalizationType (class in dorothy.config.schema) normalize_dense_data() (in module dorothy.inference.evaluation_utils) normalize_single_survey_data() (in module dorothy.inference.evaluation_utils) normalize_sparse_data() (in module dorothy.inference.evaluation_utils) normalize_spectra() (in module dorothy.data.fits_loader) normalize_spectrum() (in module dorothy.data.fits_loader) normalized (dorothy.inference.evaluation_utils.HeadEvaluation attribute) normalized units normalizer (dorothy.inference.checkpoint.LoadedCheckpoint attribute) (dorothy.inference.predictor.Predictor attribute) (dorothy.training.trainer.Trainer attribute)

O

optimizer (dorothy.training.trainer.Trainer attribute) OptimizerConfig (class in dorothy.config.schema) OptimizerType (class in dorothy.config.schema) output_dim (dorothy.models.multi_head_mlp.SharedTrunk attribute) output_features (dorothy.config.schema.MultiHeadModelConfig property) (dorothy.models.mlp.MLP attribute)

output_head (dorothy.models.multi_head_mlp.MultiHeadMLP attribute) output_heads (dorothy.models.multi_head_mlp.MultiHeadMLP attribute) output_modules() (in module dorothy.models.output) OutputHead (class in dorothy.models.multi_head_mlp) overall (dorothy.inference.evaluator.SurveyEvaluationResult attribute)

P

p_block_max (dorothy.data.augmentation.DynamicInputMasking attribute) p_block_min (dorothy.data.augmentation.DynamicInputMasking attribute) p_label_max (dorothy.data.augmentation.DynamicLabelMasking attribute) p_label_min (dorothy.data.augmentation.DynamicLabelMasking attribute) p_labelset_max (dorothy.data.augmentation.DynamicLabelMasking attribute) p_labelset_min (dorothy.data.augmentation.DynamicLabelMasking attribute) p_survey_max (dorothy.data.augmentation.DynamicInputMasking attribute) p_survey_min (dorothy.data.augmentation.DynamicInputMasking attribute) parameter_display_name() (in module dorothy.config.schema) parameter_names (dorothy.analysis.label_comparison.LabelComparison attribute) (dorothy.analysis.saliency.AblationSaliencyAnalyzer attribute) (dorothy.analysis.saliency.AblationSaliencyResult attribute) (dorothy.analysis.saliency.SaliencyAnalyzer attribute) (dorothy.analysis.saliency.SaliencyResult attribute) (dorothy.config.schema.ExperimentConfig property) (dorothy.data.catalogue_loader.CatalogueInfo attribute) (dorothy.data.fits_loader.SpectralData attribute) (dorothy.inference.checkpoint.LoadedCheckpoint property) (dorothy.inference.evaluation_utils.HeadEvaluation attribute) (dorothy.inference.evaluator.EvaluationResult attribute) (dorothy.inference.evaluator.Evaluator attribute) (dorothy.inference.predictor.PredictionResult attribute) (dorothy.inference.predictor.Predictor attribute) (dorothy.training.trainer.Trainer attribute) (dorothy.training.trainer.TrainingHistory attribute) ParameterMetrics (class in dorothy.inference.evaluator) parameters (dorothy.data.fits_loader.FITSLoader attribute) (dorothy.data.normalizer.LabelNormalizer attribute) ParameterStats (class in dorothy.data.normalizer) path (dorothy.inference.checkpoint.LoadedCheckpoint attribute) percentile (dorothy.analysis.knn_anomaly.AnomalyDetector attribute) (dorothy.analysis.knn_anomaly.AnomalyResult attribute) physical (dorothy.inference.evaluation_utils.HeadEvaluation attribute) plot_ablation_parameter_saliency() (in module dorothy.analysis.saliency) plot_ablation_saliency_heatmap() (in module dorothy.analysis.saliency) plot_grokking_metrics() (in module dorothy.visualization.training_plots) plot_histograms() (in module dorothy.visualization.evaluation_plots) plot_label_differences() (in module dorothy.analysis.label_comparison)

plot_loss_components() (in module dorothy.visualization.training_plots) plot_loss_curves() (in module dorothy.visualization.training_plots) plot_metrics_evolution() (in module dorothy.visualization.training_plots) plot_parameter_saliency() (in module dorothy.analysis.saliency) plot_per_labelset_loss_curves() (in module dorothy.visualization.training_plots) plot_per_labelset_metrics() (in module dorothy.visualization.training_plots) plot_per_labelset_metrics_evolution() (in module dorothy.visualization.training_plots) plot_per_survey_loss_curves() (in module dorothy.visualization.training_plots) plot_per_survey_metrics() (in module dorothy.visualization.training_plots) plot_per_survey_metrics_evolution() (in module dorothy.visualization.training_plots) plot_pred_vs_truth() (in module dorothy.visualization.evaluation_plots) plot_residuals() (in module dorothy.visualization.evaluation_plots) plot_saliency_heatmap() (in module dorothy.analysis.saliency) plot_val_metrics() (in module dorothy.visualization.training_plots) plot_zscore_calibration() (in module dorothy.visualization.training_plots) plot_zscore_histograms() (in module dorothy.visualization.evaluation_plots) pred_unc_p16 (dorothy.inference.evaluator.ParameterMetrics attribute) pred_unc_p50 (dorothy.inference.evaluator.ParameterMetrics attribute) pred_unc_p84 (dorothy.inference.evaluator.ParameterMetrics attribute) predict() (dorothy.inference.predictor.Predictor method) (dorothy.training.trainer.Trainer method) predict_all_chunked() (dorothy.inference.predictor.Predictor method) predict_chunked() (dorothy.inference.predictor.Predictor method) predict_from_fits() (in module dorothy.inference.predictor) predict_multi_labelset() (dorothy.training.trainer.Trainer method) predicted scatter predicted_scatter() (in module dorothy.losses.heteroscedastic) predicted_spread (dorothy.analysis.label_comparison.LabelComparison attribute) prediction PredictionResult (class in dorothy.inference.predictor) predictions (dorothy.analysis.saliency.AblationSaliencyResult attribute) (dorothy.analysis.saliency.SaliencyResult attribute) (dorothy.inference.predictor.PredictionResult attribute) Predictor (class in dorothy.inference.predictor) predictor (dorothy.analysis.knn_anomaly.AnomalyDetector attribute) primary_errors (dorothy.data.catalogue_loader.MergedCatalogueData attribute) primary_flags (dorothy.data.catalogue_loader.MergedCatalogueData attribute) primary_labels (dorothy.data.catalogue_loader.MergedCatalogueData attribute)

Q

quality_mask (dorothy.data.fits_loader.SpectralData attribute)

R

ra (dorothy.analysis.saliency.AblationSaliencyResult attribute) (dorothy.analysis.saliency.SaliencyResult attribute) (dorothy.data.catalogue_loader.CatalogueData attribute) (dorothy.data.catalogue_loader.MergedCatalogueData attribute) (dorothy.data.catalogue_loader.SparseMergedData attribute) ratio (dorothy.analysis.label_comparison.LabelComparison attribute) raw_output (dorothy.inference.predictor.PredictionResult attribute)

reduction (dorothy.losses.heteroscedastic.HeteroscedasticLoss attribute) residual_blocks() (in module dorothy.models.utils) residual_stream (dorothy.models.multi_head_mlp.MultiHeadMLP attribute) ResidualBlock (class in dorothy.models.utils) resolve_device() (in module dorothy.utils) rmse (dorothy.inference.evaluator.ParameterMetrics attribute) robust_scatter (dorothy.inference.evaluator.ParameterMetrics attribute) robust_spread (dorothy.analysis.label_comparison.LabelComparison attribute)

S

SaliencyAnalyzer (class in dorothy.analysis.saliency) SaliencyResult (class in dorothy.analysis.saliency) save() (dorothy.analysis.knn_anomaly.AnomalyDetector method) (dorothy.data.normalizer.LabelNormalizer method) (dorothy.data.training_data.DataSplit method) (dorothy.training.trainer.TrainingHistory method) save_calibration() (in module dorothy.inference.calibration) save_checkpoint() (dorothy.training.trainer.Trainer method) save_learning_rates() (dorothy.training.trainer.TrainingHistory method) save_yaml() (dorothy.config.schema.ExperimentConfig method) scaler (dorothy.training.trainer.Trainer attribute) scatter (dorothy.models.output.LabelAwareOutput attribute) scatter floor scatter_floor (dorothy.analysis.saliency.AblationSaliencyAnalyzer attribute) (dorothy.analysis.saliency.SaliencyAnalyzer attribute) (dorothy.inference.checkpoint.LoadedCheckpoint property) (dorothy.inference.evaluator.Evaluator attribute) (dorothy.inference.predictor.Predictor attribute) (dorothy.losses.heteroscedastic.HeteroscedasticLoss attribute) scheduler (dorothy.training.trainer.Trainer attribute) SchedulerConfig (class in dorothy.config.schema) SchedulerType (class in dorothy.config.schema) sd (dorothy.inference.evaluator.ParameterMetrics attribute) seed (dorothy.data.augmentation.DynamicInputMasking attribute) (dorothy.data.augmentation.DynamicLabelMasking attribute) SharedTrunk (class in dorothy.models.multi_head_mlp) sigma (dorothy.data.catalogue_loader.SparseMergedData attribute) snr (dorothy.data.catalogue_loader.CatalogueData attribute) (dorothy.data.catalogue_loader.SparseMergedData attribute) source_a (dorothy.analysis.label_comparison.LabelComparison attribute) source_b (dorothy.analysis.label_comparison.LabelComparison attribute) sparse (dorothy.data.training_data.TrainingData attribute) SparseBatchSource (class in dorothy.training.batches)

SparseMergedData (class in dorothy.data.catalogue_loader) spectral_error (dorothy.data.fits_loader.SpectralData attribute) spectral_mask (dorothy.data.fits_loader.SpectralData attribute) SpectralData (class in dorothy.data.fits_loader) spectrum (dorothy.analysis.saliency.AblationSaliencyResult attribute) (dorothy.analysis.saliency.SaliencyResult attribute) spectrum_error (dorothy.analysis.saliency.AblationSaliencyResult attribute) (dorothy.analysis.saliency.SaliencyResult attribute) split split() (dorothy.data.training_data.TrainingData method) stats (dorothy.data.normalizer.LabelNormalizer attribute) summary() (dorothy.analysis.label_comparison.LabelComparison method) (dorothy.inference.evaluator.EvaluationResult method) (dorothy.inference.evaluator.SurveyEvaluationResult method) super-catalogue survey (dorothy.analysis.saliency.AblationSaliencyResult attribute) (dorothy.analysis.saliency.SaliencyResult attribute) (dorothy.inference.predictor.Predictor attribute) survey_name (dorothy.data.catalogue_loader.CatalogueData attribute) (dorothy.inference.evaluator.EvaluationResult attribute) (dorothy.models.multi_head_mlp.SurveyEncoder attribute) survey_names (dorothy.config.schema.MultiHeadModelConfig property) (dorothy.data.catalogue_loader.CatalogueInfo attribute) (dorothy.data.catalogue_loader.MergedCatalogueData property) (dorothy.inference.evaluator.SurveyEvaluationResult attribute) (dorothy.models.multi_head_mlp.MultiHeadMLP attribute) survey_sample_counts (dorothy.inference.evaluator.SurveyEvaluationResult attribute) SurveyEncoder (class in dorothy.models.multi_head_mlp) SurveyEvaluationResult (class in dorothy.inference.evaluator) surveys (dorothy.data.catalogue_loader.CatalogueInfo attribute) (dorothy.data.catalogue_loader.MergedCatalogueData attribute) (dorothy.data.catalogue_loader.SparseMergedData attribute)

T

teff_in_log (dorothy.inference.evaluator.Evaluator attribute) TensorBatchSource (class in dorothy.training.batches) test (dorothy.data.training_data.DataSplit attribute) test_ratio (dorothy.config.schema.DataConfig property) threshold (dorothy.analysis.knn_anomaly.AnomalyDetector attribute) (dorothy.analysis.knn_anomaly.AnomalyResult attribute) to_dict() (dorothy.inference.evaluation_utils.HeadEvaluation method) (dorothy.inference.evaluator.EvaluationResult method) (dorothy.inference.evaluator.ParameterMetrics method) (dorothy.inference.evaluator.SurveyEvaluationResult method) (dorothy.inference.predictor.PredictionResult method)

total_time (dorothy.training.trainer.TrainingHistory attribute) train (dorothy.data.training_data.DataSplit attribute) train_embeddings (dorothy.analysis.knn_anomaly.AnomalyDetector attribute) train_losses (dorothy.training.trainer.TrainingHistory attribute) Trainer (class in dorothy.training.trainer) TrainingConfig (class in dorothy.config.schema) TrainingData (class in dorothy.data.training_data) TrainingHistory (class in dorothy.training.trainer) transform() (dorothy.data.normalizer.LabelNormalizer method) transform_labels() (dorothy.data.normalizer.LabelNormalizer method) trunk (dorothy.models.multi_head_mlp.MultiHeadMLP attribute)

U

uncalibrated (dorothy.inference.evaluation_utils.HeadEvaluation attribute) uncertainties (dorothy.analysis.saliency.AblationSaliencyResult attribute) (dorothy.analysis.saliency.SaliencyResult attribute) (dorothy.inference.predictor.PredictionResult attribute)

uncertainty update (dorothy.models.utils.ResidualBlock attribute) use_log (dorothy.data.normalizer.ParameterStats attribute)

V

val (dorothy.data.training_data.DataSplit attribute) val_loss_breakdown (dorothy.training.trainer.TrainingHistory attribute) val_losses (dorothy.training.trainer.TrainingHistory attribute) val_metrics (dorothy.training.trainer.TrainingHistory attribute) valid_indices (dorothy.data.training_data.TrainingData attribute) validate_ema() (dorothy.config.schema.TrainingConfig method) validate_head_dropout() (dorothy.config.schema.MultiHeadModelConfig class method) validate_hidden_layers() (dorothy.config.schema.ModelConfig class method)

validate_label_source_weights() (dorothy.config.schema.TrainingConfig class method) validate_lr_range() (dorothy.config.schema.SchedulerConfig method) validate_model_config() (dorothy.config.schema.ExperimentConfig method) validate_parameter_names() (dorothy.config.schema.DataConfig class method) validate_ranges() (dorothy.config.schema.InputMaskingConfig method) (dorothy.config.schema.LabelMaskingConfig method) validate_ratios() (dorothy.config.schema.DataConfig method) validate_survey_wavelengths() (dorothy.config.schema.MultiHeadModelConfig class method) version (dorothy.data.catalogue_loader.CatalogueInfo attribute)

W

wavelength (dorothy.analysis.saliency.AblationSaliencyResult attribute) (dorothy.analysis.saliency.SaliencyResult attribute) (dorothy.data.catalogue_loader.CatalogueData attribute) (dorothy.data.fits_loader.SpectralData attribute)

wavelengths (dorothy.data.catalogue_loader.SparseMergedData attribute) weight_norms (dorothy.training.trainer.TrainingHistory attribute) weight_updates (dorothy.training.trainer.TrainingHistory attribute)

Z

z-score z_label_errors (dorothy.inference.evaluation_utils.HeadEvaluation attribute)

z_median (dorothy.inference.evaluator.ParameterMetrics attribute) z_robust_scatter (dorothy.inference.evaluator.ParameterMetrics attribute) zero_invalid_pixels() (in module dorothy.data.fits_loader)