Aquila-Consortium/csiborgtools
0
0
Fork 0
mirror of https://github.com/Richard-Sti/csiborgtools.git synced 2025-04-19 06:40:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add void mock

Browse source
This commit is contained in:
rstiskalek 2024-10-07 11:51:41 +01:00
parent d7da107d1c
commit 225bd6eabb
1 changed files with 9 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
scripts/flow/flow_validation.py
View file

@ -120,6 +120,8 @@ def get_models(ksim, get_model_kwargs, mag_selection, void_kwargs,
fpath = join(folder, "PV/CF4/CF4_TF-distances.hdf5") fpath = join(folder, "PV/CF4/CF4_TF-distances.hdf5")
elif cat in ["CF4_GroupAll"]: elif cat in ["CF4_GroupAll"]:
fpath = join(folder, "PV/CF4/CF4_GroupAll.hdf5") fpath = join(folder, "PV/CF4/CF4_GroupAll.hdf5")
elif "IndranilVoidTFRMock" in cat:
fpath = None
else: else:
raise ValueError(f"Unsupported catalogue: `{ARGS.catalogue}`.") raise ValueError(f"Unsupported catalogue: `{ARGS.catalogue}`.")
@ -135,14 +137,6 @@ def get_models(ksim, get_model_kwargs, mag_selection, void_kwargs,
return models return models
def select_void_h(kind):
hs = {"mb": 0.7615, "gauss": 0.7724, "exp": 0.7725}
try:
return hs[kind]
except KeyError:
raise ValueError(f"Unknown void kind: `{kind}`.")
def get_harmonic_evidence(samples, log_posterior, nchains_harmonic, epoch_num): def get_harmonic_evidence(samples, log_posterior, nchains_harmonic, epoch_num):
"""Compute evidence using the `harmonic` package.""" """Compute evidence using the `harmonic` package."""
data, names = csiborgtools.dict_samples_to_array(samples) data, names = csiborgtools.dict_samples_to_array(samples)
@ -244,7 +238,7 @@ def get_distmod_hyperparams(catalogue, sample_alpha, sample_mag_dipole):
"alpha_min": alpha_min, "alpha_max": alpha_max, "alpha_min": alpha_min, "alpha_max": alpha_max,
"sample_alpha": sample_alpha "sample_alpha": sample_alpha
} }
elif catalogue in ["SFI_gals", "2MTF"] or "CF4_TFR" in catalogue: elif catalogue in ["SFI_gals", "2MTF"] or "CF4_TFR" in catalogue or "IndranilVoidTFRMock" in catalogue: # noqa
return {"e_mu_min": 0.001, "e_mu_max": 1.0, return {"e_mu_min": 0.001, "e_mu_max": 1.0,
"a_mean": -21., "a_std": 5.0, "a_mean": -21., "a_std": 5.0,
"b_mean": -5.95, "b_std": 4.0, "b_mean": -5.95, "b_std": 4.0,
@ -300,10 +294,10 @@ if __name__ == "__main__":
########################################################################### ###########################################################################
# `None` means default behaviour # `None` means default behaviour
nsteps = 1_000 nsteps = 2_000
nburn = 1_000 nburn = 2_000
zcmb_min = None zcmb_min = None
zcmb_max = 0.04 zcmb_max = 0.05
nchains_harmonic = 10 nchains_harmonic = 10
num_epochs = 50 num_epochs = 50
inference_method = "mike" inference_method = "mike"
@ -314,7 +308,7 @@ if __name__ == "__main__":
sample_Vmag_vax = False sample_Vmag_vax = False
sample_Vmono = False sample_Vmono = False
sample_mag_dipole = False sample_mag_dipole = False
wo_num_dist_marginalisation = True wo_num_dist_marginalisation = False
absolute_calibration = None absolute_calibration = None
calculate_harmonic = (False if inference_method == "bayes" else True) and (not wo_num_dist_marginalisation) # noqa calculate_harmonic = (False if inference_method == "bayes" else True) and (not wo_num_dist_marginalisation) # noqa
sample_h = True if absolute_calibration is not None else False sample_h = True if absolute_calibration is not None else False
@ -361,7 +355,7 @@ if __name__ == "__main__":
"`IndranilVoid` does not have multiple realisations.") "`IndranilVoid` does not have multiple realisations.")
profile = ARGS.simname.split("_")[-1] profile = ARGS.simname.split("_")[-1]
h = select_void_h(profile) h = csiborgtools.flow.select_void_h(profile)
rdist = np.arange(0, 165, 0.5) rdist = np.arange(0, 165, 0.5)
void_kwargs = {"profile": profile, "h": h, "order": 1, "rdist": rdist} void_kwargs = {"profile": profile, "h": h, "order": 1, "rdist": rdist}
else: else:
@ -380,7 +374,7 @@ if __name__ == "__main__":
calibration_hyperparams = {"Vext_min": -3000, "Vext_max": 3000, calibration_hyperparams = {"Vext_min": -3000, "Vext_max": 3000,
"Vmono_min": -1000, "Vmono_max": 1000, "Vmono_min": -1000, "Vmono_max": 1000,
"beta_min": -10.0, "beta_max": 10.0, "beta_min": -10.0, "beta_max": 10.0,
"sigma_v_min": 1.0, "sigma_v_max": 5000 if "IndranilVoid_" in ARGS.simname else 750., # noqa "sigma_v_min": 1.0, "sigma_v_max": 1000 if "IndranilVoid_" in ARGS.simname else 750., # noqa
"h_min": 0.01, "h_max": 5.0, "h_min": 0.01, "h_max": 5.0,
"no_Vext": False if no_Vext is None else no_Vext, # noqa "no_Vext": False if no_Vext is None else no_Vext, # noqa
"sample_Vmag_vax": sample_Vmag_vax, "sample_Vmag_vax": sample_Vmag_vax,