Skip to content
Snippets Groups Projects
Select Git revision
  • Binary
  • add-version-information
  • master default protected
  • os-path-join
  • develop-GA
  • timeFstatmap
  • add-higher-spindown-components
  • develop-DK
  • adds-header-to-grid-search
  • v1.3
  • v1.2
  • v1.1.2
  • v1.1.0
  • v1.0.1
14 results

short_transient_search_MCMC.py

Blame
  • Forked from Gregory Ashton / PyFstat
    6 commits behind the upstream repository.
    short_transient_search_MCMC.py 1.24 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.0, "upper": F0 + DeltaF0 / 2.0},
        "F1": {"type": "unif", "lower": F1 - DeltaF1 / 2.0, "upper": F1 + DeltaF1 / 2.0},
        "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()