Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | P | R | S | T | U | V _ __call__() (n2j.losses.DiagonalGaussianNLL method) (n2j.losses.DoubleGaussianNLL method) (n2j.losses.FullRankGaussianNLL method) (n2j.losses.gaussian_nll.DiagonalGaussianNLL method) (n2j.losses.gaussian_nll.DoubleGaussianNLL method) (n2j.losses.gaussian_nll.FullRankGaussianNLL method) (n2j.losses.local_global_loss.MSELoss method) (n2j.losses.MSELoss method) (n2j.losses.nll.DiagonalGaussianNLL method) (n2j.losses.nll.DoubleGaussianNLL method) (n2j.losses.nll.FullRankGaussianNLL method) (transform_utils.ComposeXYLocal method) (transform_utils.MagErrorSimulator method) (transform_utils.MagErrorSimulatorTorch method) (transform_utils.Metadata method) (transform_utils.Rejector method) (transform_utils.Slicer method) (transform_utils.Standardizer method) __getitem__() (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2Graph method) __repr__() (n2j.trainer.Trainer method) _define_paths() (base_raytracer.BaseRaytracer method) _format_input_params() (transform_utils.MagErrorSimulator method) (transform_utils.MagErrorSimulatorTorch method) _get_pointings() (cosmodc2_raytracer.CosmoDC2Raytracer method) _get_pointings_on_grid() (cosmodc2_raytracer.CosmoDC2Raytracer method) _get_sightlines() (n2j.trainval_data.graphs.base_graph.BaseGraph method) _get_y_class() (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix method) _log_kappa_recovery() (n2j.trainer.Trainer method) _log_kappa_recovery_flow() (n2j.trainer.Trainer method) _save_graph_to_disk() (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix method) _set_column_names() (cosmodc2_raytracer.CosmoDC2Raytracer method) _slice_input_params() (transform_utils.MagErrorSimulator method) (transform_utils.MagErrorSimulatorTorch method) A apply_calibration() (base_raytracer.BaseRaytracer method) approx_mean_kappa() (in module raytracing_utils) B bands (transform_utils.MagErrorSimulator attribute) (transform_utils.MagErrorSimulatorTorch attribute) base_raytracer module BaseGraph (class in n2j.trainval_data.graphs.base_graph) BaseRaytracer (class in base_raytracer) bnn_kappa_path (inference_manager.InferenceManager property) C C_ms (transform_utils.MagErrorSimulator attribute) (transform_utils.MagErrorSimulatorTorch attribute) calculate_5sigma_depths() (transform_utils.MagErrorSimulator method) (transform_utils.MagErrorSimulatorTorch method) calculate_delta_C_ms() (transform_utils.MagErrorSimulator method) (transform_utils.MagErrorSimulatorTorch method) calculate_rand_err() (transform_utils.MagErrorSimulator method) (transform_utils.MagErrorSimulatorTorch method) calibration module COLUMN_NAMING (cosmodc2_raytracer.CosmoDC2Raytracer attribute) columns (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix attribute) ComposeXYLocal (class in transform_utils) compute_metrics() (inference_manager.InferenceManager method) configure_model() (inference_manager.InferenceManager method) (n2j.trainer.Trainer method) configure_optim() (n2j.trainer.Trainer method) coord_utils module cosmodc2_raytracer module CosmoDC2Graph (class in n2j.trainval_data.graphs.cosmodc2_graph) CosmoDC2GraphHealpix (class in n2j.trainval_data.graphs.cosmodc2_graph) CosmoDC2Raytracer (class in cosmodc2_raytracer) D data_stats() (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2Graph method) data_stats_valtest() (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2Graph method) DEBUG (in module infer_utils) (in module n2j.models.flow) delete_previous() (inference_manager.InferenceManager method) delta_C_m_infs (transform_utils.MagErrorSimulator attribute) (transform_utils.MagErrorSimulatorTorch attribute) DiagonalGaussianNLL (class in n2j.losses) (class in n2j.losses.gaussian_nll) (class in n2j.losses.nll) DoubleGaussianNLL (class in n2j.losses) (class in n2j.losses.gaussian_nll) (class in n2j.losses.nll) download() (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix method) E estimate_mean_kappas() (base_raytracer.BaseRaytracer method) export_stats() (summary_stats_baseline.SummaryStats method) F fit_gp() (in module raytracing_utils) fit_kde_on_weighted_samples() (in module infer_utils) Flow (class in n2j.models.flow) fmualpha() (n2j.models.flow.MAF method) forward() (n2j.models.flow.Flow method) (n2j.models.flow.MADE method) (n2j.models.flow.MAF method) (n2j.models.flow.MaskedLinear method) (n2j.models.flow.Perm method) (n2j.models.GATNet method) (n2j.models.GCNNet method) (n2j.models.gnn.GATNet method) (n2j.models.gnn.GCNNet method) (n2j.models.gnn.GravNet method) (n2j.models.gnn.SageNet method) (n2j.models.GravNet method) (n2j.models.N2JNet method) (n2j.models.n2jnet.N2JNet method) (n2j.models.SageNet method) FullRankGaussianNLL (class in n2j.losses) (class in n2j.losses.gaussian_nll) (class in n2j.losses.nll) G gammas (transform_utils.MagErrorSimulator attribute) (transform_utils.MagErrorSimulatorTorch attribute) GATNet (class in n2j.models) (class in n2j.models.gnn) GCNNet (class in n2j.models) (class in n2j.models.gnn) get() (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix method) get_bands_in_x() (in module transform_utils) get_bnn_kappa() (inference_manager.InferenceManager method) get_calibration_plot() (inference_manager.InferenceManager method) get_chain_plot() (in module infer_utils) get_concentration() (in module halo_utils) get_config() (in module n2j.config_utils) get_config_modular() (in module n2j.config_utils) get_corners() (in module coord_utils) get_distance() (in module coord_utils) get_edges() (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix method) get_gals_iterator() (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix method) get_halos_iterator() (base_raytracer.BaseRaytracer method) (cosmodc2_raytracer.CosmoDC2Raytracer method) get_healpix_centers() (in module coord_utils) get_idx() (in module transform_utils) get_inv_dist_number_counts() (in module summary_stats_baseline) get_kappa_log_weights() (in module infer_utils) (inference_manager.InferenceManager method) get_kappa_log_weights_grid() (inference_manager.InferenceManager method) get_kappa_log_weights_vectorized() (in module infer_utils) get_log_p_k_given_omega_int() (inference_manager.InferenceManager method) get_log_p_k_given_omega_int_analytic() (in module infer_utils) get_log_p_k_given_omega_int_kde() (in module infer_utils) get_log_p_k_given_omega_int_loop() (inference_manager.InferenceManager method) get_log_p_k_given_omega_int_per_los() (inference_manager.InferenceManager method) get_los_halos() (cosmodc2_raytracer.CosmoDC2Raytracer method) get_los_node() (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix method) get_mcmc_samples() (in module infer_utils) get_mean_kappas() (base_raytracer.BaseRaytracer method) get_nfw_kwargs() (in module halo_utils) get_normal_logpdf() (in module infer_utils) get_number_counts() (in module summary_stats_baseline) get_omega_post() (in module infer_utils) get_omega_post_loop() (in module infer_utils) get_omega_samples() (inference_manager.InferenceManager method) get_optimal_threshold() (in module summary_stats_baseline) get_overview_table() (summary_stats_baseline.Matcher method) get_p_Y_val_approx_mahalanobis() (in module calibration) get_padded_nside() (in module coord_utils) get_pointings_iterator() (base_raytracer.BaseRaytracer method) (cosmodc2_raytracer.CosmoDC2Raytracer method) get_reweighted_bnn_kappa() (inference_manager.InferenceManager method) get_samples() (summary_stats_baseline.Matcher method) get_sigmas() (transform_utils.MagErrorSimulator method) (transform_utils.MagErrorSimulatorTorch method) get_skycoord() (in module coord_utils) get_summary_stats() (inference_manager.InferenceManager method) get_target_nside() (in module coord_utils) get_true_kappa() (inference_manager.InferenceManager method) global_loss() (n2j.models.N2JNet method) (n2j.models.n2jnet.N2JNet method) GravNet (class in n2j.models) (class in n2j.models.gnn) H halo_utils module I include_los (inference_manager.InferenceManager property) infer() (n2j.trainer.Trainer method) infer_utils module inference_manager module InferenceManager (class in inference_manager) invert() (n2j.models.flow.Flow method) (n2j.models.flow.MAF method) (n2j.models.flow.Perm method) is_decreasing() (in module n2j.trainer) is_inside() (in module coord_utils) K k_ms (transform_utils.MagErrorSimulator attribute) (transform_utils.MagErrorSimulatorTorch attribute) KAPPA_DIFF (cosmodc2_raytracer.CosmoDC2Raytracer attribute) L len() (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix method) LENSING_NSIDE (cosmodc2_raytracer.CosmoDC2Raytracer attribute) load_context() (n2j.models.flow.MAF method) load_dataset() (inference_manager.InferenceManager method) (n2j.trainer.Trainer method) load_mean_kappas_from_file() (base_raytracer.BaseRaytracer method) load_state() (inference_manager.InferenceManager method) (n2j.trainer.Trainer method) local_loss() (n2j.models.N2JNet method) (n2j.models.n2jnet.N2JNet method) log_p_k_given_omega_int_path (inference_manager.InferenceManager property) log_prob_mcmc() (in module infer_utils) log_prob_mcmc_loop() (in module infer_utils) loss() (n2j.models.N2JNet method) (n2j.models.n2jnet.N2JNet method) M m_skys (transform_utils.MagErrorSimulator attribute) (transform_utils.MagErrorSimulatorTorch attribute) MADE (class in n2j.models.flow) MAF (class in n2j.models.flow) MagErrorSimulator (class in transform_utils) MagErrorSimulatorTorch (class in transform_utils) MaskedLinear (class in n2j.models.flow) match() (in module coord_utils) (in module summary_stats_baseline) match_summary_stats() (summary_stats_baseline.Matcher method) Matcher (class in summary_stats_baseline) matching_dir (inference_manager.InferenceManager property) Metadata (class in transform_utils) module base_raytracer calibration coord_utils cosmodc2_raytracer halo_utils infer_utils inference_manager n2j n2j.config_utils n2j.data n2j.losses n2j.losses.gaussian_nll n2j.losses.local_global_loss n2j.losses.nll n2j.models n2j.models.flow n2j.models.gnn n2j.models.n2jnet n2j.run_raytracing n2j.tests n2j.tests.test_config_utils n2j.trainer n2j.trainval_data n2j.trainval_data.graphs n2j.trainval_data.graphs.base_graph n2j.trainval_data.graphs.cosmodc2_graph raytracing_utils running_stats summary_stats_baseline test_coord_utils test_halo_utils test_infer_utils test_inference_manager test_running_stats test_summary_stats test_transform_utils transform_utils MSELoss (class in n2j.losses) (class in n2j.losses.local_global_loss) N n2j module n2j.config_utils module n2j.data module n2j.losses module n2j.losses.gaussian_nll module n2j.losses.local_global_loss module n2j.losses.nll module n2j.models module n2j.models.flow module n2j.models.gnn module n2j.models.n2jnet module n2j.run_raytracing module n2j.tests module n2j.tests.test_config_utils module n2j.trainer module n2j.trainval_data module n2j.trainval_data.graphs module n2j.trainval_data.graphs.base_graph module n2j.trainval_data.graphs.cosmodc2_graph module N2JNet (class in n2j.models) (class in n2j.models.n2jnet) n_features (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix property) n_test (inference_manager.InferenceManager property) NSIDE (cosmodc2_raytracer.CosmoDC2Raytracer attribute) num_visits_10_year (transform_utils.MagErrorSimulator attribute) (transform_utils.MagErrorSimulatorTorch attribute) P parallel_raytrace() (cosmodc2_raytracer.CosmoDC2Raytracer method) parser (in module n2j.run_raytracing) Perm (class in n2j.models.flow) plot_calibration() (in module calibration) post_reweighting_metrics (inference_manager.InferenceManager property) post_reweighting_metrics_grid (inference_manager.InferenceManager property) post_reweighting_metrics_grid_path (inference_manager.InferenceManager property) post_reweighting_metrics_path (inference_manager.InferenceManager property) posterior_name (n2j.losses.DiagonalGaussianNLL attribute) (n2j.losses.DoubleGaussianNLL attribute) (n2j.losses.FullRankGaussianNLL attribute) (n2j.losses.gaussian_nll.DiagonalGaussianNLL attribute) (n2j.losses.gaussian_nll.DoubleGaussianNLL attribute) (n2j.losses.gaussian_nll.FullRankGaussianNLL attribute) (n2j.losses.nll.DiagonalGaussianNLL attribute) (n2j.losses.nll.DoubleGaussianNLL attribute) (n2j.losses.nll.FullRankGaussianNLL attribute) pre_reweighting_metrics (inference_manager.InferenceManager property) pre_reweighting_metrics_path (inference_manager.InferenceManager property) process() (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix method) process_single() (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix method) processed_file_fmt (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix property) processed_file_names (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix property) processed_file_path_fmt (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix property) R raw_dir (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix property) raw_file_name (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix property) raw_file_names (n2j.trainval_data.graphs.cosmodc2_graph.CosmoDC2GraphHealpix property) raytracing_utils module Rejector (class in transform_utils) resample_from_pdf() (in module infer_utils) resample_from_samples() (in module infer_utils) reweighted_bnn_kappa_grid_path (inference_manager.InferenceManager property) reweighted_bnn_kappa_per_sample_path (inference_manager.InferenceManager property) reweighted_grid_dir (inference_manager.InferenceManager property) reweighted_per_sample_dir (inference_manager.InferenceManager property) run_mcmc() (in module infer_utils) run_mcmc_for_omega_post() (inference_manager.InferenceManager method) run_mcmc_for_omega_post_summary_stats() (inference_manager.InferenceManager method) running_stats module RunningStats (class in running_stats) S SageNet (class in n2j.models) (class in n2j.models.gnn) sample() (n2j.losses.DiagonalGaussianNLL method) (n2j.losses.DoubleGaussianNLL method) (n2j.losses.FullRankGaussianNLL method) (n2j.losses.gaussian_nll.DiagonalGaussianNLL method) (n2j.losses.gaussian_nll.DoubleGaussianNLL method) (n2j.losses.gaussian_nll.FullRankGaussianNLL method) (n2j.losses.nll.DiagonalGaussianNLL method) (n2j.losses.nll.DoubleGaussianNLL method) (n2j.losses.nll.FullRankGaussianNLL method) sample_in_aperture() (in module coord_utils) sample_kappas() (cosmodc2_raytracer.CosmoDC2Raytracer method) save_state() (n2j.trainer.Trainer method) seed_everything() (inference_manager.InferenceManager method) (n2j.trainer.Trainer method) seeings (transform_utils.MagErrorSimulator attribute) (transform_utils.MagErrorSimulatorTorch attribute) set_mask() (n2j.models.flow.MaskedLinear method) set_stats() (summary_stats_baseline.SummaryStats method) set_trained_pred() (n2j.losses.DiagonalGaussianNLL method) (n2j.losses.DoubleGaussianNLL method) (n2j.losses.FullRankGaussianNLL method) (n2j.losses.gaussian_nll.DiagonalGaussianNLL method) (n2j.losses.gaussian_nll.DoubleGaussianNLL method) (n2j.losses.gaussian_nll.FullRankGaussianNLL method) (n2j.losses.nll.DiagonalGaussianNLL method) (n2j.losses.nll.DoubleGaussianNLL method) (n2j.losses.nll.FullRankGaussianNLL method) set_valtest_loading() (inference_manager.InferenceManager method) setUpClass() (n2j.tests.test_config_utils.TestConfigUtils class method) (test_coord_utils.TestCoordUtils class method) (test_halo_utils.TestHaloUtils class method) (test_infer_utils.TestInferUtils class method) (test_inference_manager.TestInferenceManager class method) (test_running_stats.TestRunningStats class method) (test_summary_stats.TestSummaryStats class method) (test_transform_utils.TestMagErrorSimulator class method) (test_transform_utils.TestMagErrorSimulatorTorch class method) (test_transform_utils.TestRejector class method) single_raytrace() (cosmodc2_raytracer.CosmoDC2Raytracer method) slice() (n2j.losses.DiagonalGaussianNLL method) (n2j.losses.DoubleGaussianNLL method) (n2j.losses.FullRankGaussianNLL method) (n2j.losses.gaussian_nll.DiagonalGaussianNLL method) (n2j.losses.gaussian_nll.DoubleGaussianNLL method) (n2j.losses.gaussian_nll.FullRankGaussianNLL method) (n2j.losses.nll.DiagonalGaussianNLL method) (n2j.losses.nll.DoubleGaussianNLL method) (n2j.losses.nll.FullRankGaussianNLL method) Slicer (class in transform_utils) Standardizer (class in transform_utils) Subgraph (class in n2j.trainval_data.graphs.base_graph) summary_stats_baseline module SummaryStats (class in summary_stats_baseline) T tearDownClass() (n2j.tests.test_config_utils.TestConfigUtils class method) (test_coord_utils.TestCoordUtils class method) (test_halo_utils.TestHaloUtils class method) (test_infer_utils.TestInferUtils class method) (test_inference_manager.TestInferenceManager class method) (test_summary_stats.TestSummaryStats class method) (test_transform_utils.TestMagErrorSimulator class method) (test_transform_utils.TestMagErrorSimulatorTorch class method) test_basic() (test_transform_utils.TestMagErrorSimulator method) (test_transform_utils.TestMagErrorSimulatorTorch method) test_configure_model() (test_inference_manager.TestInferenceManager method) test_coord_utils module test_correctness() (test_transform_utils.TestMetadata method) test_default() (test_transform_utils.TestRejector method) test_get_bands_in_x() (in module test_transform_utils) test_get_bnn_kappa() (test_inference_manager.TestInferenceManager method) test_get_concentration() (test_halo_utils.TestHaloUtils method) test_get_config_modular() (n2j.tests.test_config_utils.TestConfigUtils method) test_get_corners() (test_coord_utils.TestCoordUtils method) test_get_healpix_centers() (test_coord_utils.TestCoordUtils method) test_get_inv_dist_number_counts() (test_summary_stats.TestSummaryStats method) test_get_normal_logpdf() (test_infer_utils.TestInferUtils method) test_get_number_counts() (test_summary_stats.TestSummaryStats method) test_get_omega_post() (test_infer_utils.TestInferUtils method) test_get_padded_nside() (test_coord_utils.TestCoordUtils method) test_get_skycoord() (test_coord_utils.TestCoordUtils method) test_get_target_nside() (test_coord_utils.TestCoordUtils method) test_halo_utils module test_infer_utils module test_inference_manager module test_is_inside() (test_coord_utils.TestCoordUtils method) test_load_checkpoint() (test_inference_manager.TestInferenceManager method) test_load_dataset() (test_inference_manager.TestInferenceManager method) test_log_prob_mcmc_bnn() (test_infer_utils.TestInferUtils method) test_log_prob_mcmc_multiple() (test_infer_utils.TestInferUtils method) test_log_prob_mcmc_single() (test_infer_utils.TestInferUtils method) test_match() (test_coord_utils.TestCoordUtils method) test_max_only() (test_transform_utils.TestRejector method) test_min_only() (test_transform_utils.TestRejector method) test_r_band_lit() (test_transform_utils.TestMagErrorSimulator method) (test_transform_utils.TestMagErrorSimulatorTorch method) test_running_mean_np() (test_running_stats.TestRunningStats method) test_running_stats module test_running_stats() (test_running_stats.TestRunningStats method) test_running_std_np() (test_running_stats.TestRunningStats method) test_sample_in_aperture() (test_coord_utils.TestCoordUtils method) test_set_stats_export_stats() (test_summary_stats.TestSummaryStats method) test_shapes_call() (test_transform_utils.TestMagErrorSimulator method) (test_transform_utils.TestMagErrorSimulatorTorch method) test_shapes_get_sigmas() (test_transform_utils.TestMagErrorSimulator method) (test_transform_utils.TestMagErrorSimulatorTorch method) test_sigma_rand_infinite_brightness() (test_transform_utils.TestMagErrorSimulator method) (test_transform_utils.TestMagErrorSimulatorTorch method) test_sigma_rand_m_5() (test_transform_utils.TestMagErrorSimulator method) (test_transform_utils.TestMagErrorSimulatorTorch method) test_summary_stats module test_summary_stats_path (inference_manager.InferenceManager property) test_transform_utils module test_update() (test_summary_stats.TestSummaryStats method) test_upgrade_healpix() (test_coord_utils.TestCoordUtils method) TestConfigUtils (class in n2j.tests.test_config_utils) TestCoordUtils (class in test_coord_utils) TestHaloUtils (class in test_halo_utils) TestInferenceManager (class in test_inference_manager) TestInferUtils (class in test_infer_utils) TestMagErrorSimulator (class in test_transform_utils) TestMagErrorSimulatorTorch (class in test_transform_utils) TestMetadata (class in test_transform_utils) TestRejector (class in test_transform_utils) TestRunningStats (class in test_running_stats) TestSummaryStats (class in test_summary_stats) TO_200C (cosmodc2_raytracer.CosmoDC2Raytracer attribute) train() (n2j.trainer.Trainer method) train_single_epoch() (n2j.trainer.Trainer method) train_summary_stats_path (inference_manager.InferenceManager property) Trainer (class in n2j.trainer) transform_utils module true_test_kappa_path (inference_manager.InferenceManager property) true_train_kappa_path (inference_manager.InferenceManager property) TWO_PI (in module coord_utils) U uniform_around() (in module coord_utils) update() (running_stats.RunningStats method) (summary_stats_baseline.SummaryStats method) update_masks() (n2j.models.flow.MADE method) upgrade_healpix() (in module coord_utils) V visualize_kappa_post() (inference_manager.InferenceManager method) visualize_omega_post() (inference_manager.InferenceManager method)