Skip to content
Snippets Groups Projects
Select Git revision
  • 72-improve-docs-for_optimal_setup
  • master default protected
  • os-path-join
  • develop-GA
  • add-higher-spindown-components
  • v1.3
  • v1.2
  • v1.1.2
  • v1.1.0
  • v1.0.1
10 results

semi_coherent_twoglitch_search_using_MCMC.py

Blame
  • semi_coherent_twoglitch_search_using_MCMC.py 1.39 KiB
    import pyfstat
    
    F0 = 30.0
    F1 = -1e-10
    F2 = 0
    Alpha = 5e-3
    Delta = 6e-2
    tref = 362750407.0
    
    tstart = 1000000000
    duration = 100*86400
    tend = tstart + duration
    
    theta_prior = {'F0': {'type': 'norm', 'loc': F0, 'scale': abs(1e-6*F0)},
                   'F1': {'type': 'norm', 'loc': F1, 'scale': abs(1e-6*F1)},
                   'F2': F2,
                   'Alpha': Alpha,
                   'Delta': Delta,
                   'delta_F0_0': {'type': 'halfnorm', 'loc': 0,
                                  'scale': 1e-7*F0},
                   'delta_F0_1': {'type': 'halfnorm', 'loc': 0,
                                  'scale': 1e-7*F0},
                   'delta_F1_0': 0,
                   'delta_F1_1': 0,
                   'tglitch_0': {'type': 'unif',
                                 'lower': tstart+0.01*duration,
                                 'upper': tstart+0.5*duration},
                   'tglitch_1': {'type': 'unif',
                                 'lower': tstart+0.5*duration,
                                 'upper': tstart+0.99*duration},
                   }
    
    nwalkers = 100
    nsteps = [1000, 1000, 5000]
    
    mcmc = pyfstat.MCMCGlitchSearch(
        'semi_coherent_twoglitch_search', 'data', sftfilepattern='data/*twoglitch*sft',
        theta_prior=theta_prior, tref=tref, tstart=tstart,
        tend=tend, nsteps=nsteps, nwalkers=nwalkers, scatter_val=1e-10, nglitch=2)
    
    mcmc.run()
    mcmc.plot_corner(add_prior=True, tglitch_ratio=True, figsize=(10, 10))
    mcmc.print_summary()