else: if args.hd: orf = 'hd' else: orf = None gw = models.common_red_noise_block(psd=args.psd, prior=prior, Tspan=Tspan, orf=orf, gamma_val=args.gamma_gw, name='gw') model += gw log10_sigma = parameter.Uniform(-10, -4) log10_ell = parameter.Uniform(1, 4) dm_basis = linear_interp_basis_dm(dt=15 * 86400) dm_prior = se_dm_kernel(log10_sigma=log10_sigma, log10_ell=log10_ell) dm_gp = gp_signals.BasisGP(dm_prior, dm_basis, name='dm_gp') dm_block = dm_gp # Make solar wind signals print('sw_r2p ', args.sw_r2p) # if isinstance(args.sw_r2p,(float,int)): # args.sw_r2p = [args.sw_r2p] if args.sw_r2p_ranges is None: sw_r2p_ranges = args.sw_r2p elif len(args.sw_r2p) != len(args.sw_r2p_ranges): raise ValueError('Number of SW powers must match number of prior ranges!! ' 'Set # nonvarying ') else:
model_kwargs = json.load(fin) # Add to exponential dips for J1713+0747 #Model, kernel, extra DMGP, Chrom Kernel, Chrom Quad, Index model_labels = [['A', 'periodic', True, True, 'sq_exp', False, 4]] ptas = {} all_kwargs = {} # Periodic GP kernel for DM log10_sigma = parameter.Uniform(-10, -4.8) log10_ell = parameter.Uniform(1, 2.4) log10_p = parameter.Uniform(-2, -1) log10_gam_p = parameter.Uniform(-2, 2) dm_basis = gpk.linear_interp_basis_dm(dt=3 * 86400) dm_prior = gpk.periodic_kernel(log10_sigma=log10_sigma, log10_ell=log10_ell, log10_gam_p=log10_gam_p, log10_p=log10_p) dmgp = gp_signals.BasisGP(dm_prior, dm_basis, name='dm_gp1') # Periodic GP kernel for DM log10_sigma2 = parameter.Uniform(-4.8, -3) log10_ell2 = parameter.Uniform(2.4, 5) log10_p2 = parameter.Uniform(-2, 2) log10_gam_p2 = parameter.Uniform(-2, 2) dm_basis2 = gpk.linear_interp_basis_dm(dt=3 * 86400) dm_prior2 = gpk.periodic_kernel(log10_sigma=log10_sigma2,
['A', 'periodic', False, True, 'sq_exp', False, 4], ['B', 'periodic', True, True, 'sq_exp', False, 4], ['C', 'periodic', False, True, 'sq_exp', True, 4], ['D', 'periodic', True, True, 'sq_exp', True, 4], ] ptas = {} all_kwargs = {} # Periodic GP kernel for DM log10_sigma = parameter.Uniform(-4.4, -3) log10_ell = parameter.Uniform(3, 4) log10_p = parameter.Uniform(-1, 1) log10_gam_p = parameter.Uniform(-1.5, 1) dm_basis = gpk.linear_interp_basis_dm(dt=14 * 86400) dm_prior = gpk.periodic_kernel(log10_sigma=log10_sigma, log10_ell=log10_ell, log10_gam_p=log10_gam_p, log10_p=log10_p) dmgp2 = gp_signals.BasisGP(dm_prior, dm_basis, name='dm_gp2') @signal_base.function def chromatic_quad(toas, freqs, quad_coeff=np.ones(3) * 1e-10, idx=4): """ Basis for chromatic quadratic function. :param idx: index of chromatic dependence