Skip to content
Snippets Groups Projects
Select Git revision
  • master default protected
  • develop-GA
  • timeFstatmap
  • add-higher-spindown-components
  • develop-DK
  • adds-header-to-grid-search
  • v1.2
  • v1.1.2
  • v1.1.0
  • v1.0.1
10 results

short_transient_search_MCMC.py

Blame
  • Forked from Gregory Ashton / PyFstat
    50 commits behind the upstream repository.
    short_transient_search_MCMC.py 1.43 KiB
    #!/usr/bin/env python
    
    import pyfstat
    
    F0 = 30.0
    F1 = -1e-10
    F2 = 0
    Alpha = 0.5
    Delta = 1
    
    minStartTime = 1000000000
    maxStartTime = minStartTime + 2*86400
    Tspan = maxStartTime - minStartTime
    tref = minStartTime
    
    Tsft = 1800
    
    DeltaF0 = 1e-2
    DeltaF1 = 1e-9
    
    theta_prior = {'F0': {'type': 'unif',
                          'lower': F0-DeltaF0/2.,
                          'upper': F0+DeltaF0/2.},
                   'F1': {'type': 'unif',
                          'lower': F1-DeltaF1/2.,
                          'upper': F1+DeltaF1/2.},
                   'F2': F2,
                   'Alpha': Alpha,
                   'Delta': Delta,
                   'transient_tstart': {'type': 'unif',
                                        'lower': minStartTime,
                                        'upper': maxStartTime-2*Tsft},
                   'transient_duration': {'type': 'unif',
                                          'lower': 2*Tsft,
                                          'upper': Tspan-2*Tsft}
                   }
    
    ntemps = 2
    log10beta_min = -1
    nwalkers = 100
    nsteps = [100, 100]
    
    mcmc = pyfstat.MCMCTransientSearch(
        label='transient_search', outdir='data_s',
        sftfilepattern='data_s/*simulated_transient_signal*sft',
        theta_prior=theta_prior, tref=tref, minStartTime=minStartTime,
        maxStartTime=maxStartTime, nsteps=nsteps, nwalkers=nwalkers, ntemps=ntemps,
        log10beta_min=log10beta_min,
        transientWindowType='rect')
    mcmc.run()
    mcmc.plot_corner(label_offset=0.7)
    mcmc.print_summary()