def get_zvalues(fp, arg, likelihood_stats):
"""Reads the data for the z-value of the plots from the inference file.
"""
if arg == 'loglr':
zvals = likelihood_stats.loglr
zlbl = r'$\log\mathcal{L}(\vec{\vartheta})$'
elif arg == 'snr':
zvals = likelihood.snr_from_loglr(likelihood_stats.loglr)
zlbl = r'$\rho(\vec{\vartheta})$'
elif arg == 'logplr':
zvals = likelihood_stats.loglr + likelihood_stats.prior
zlbl = r'$\log[\mathcal{L}(\vec{\vartheta})p(\vec{\vartheta})]$'
elif arg == 'logposterior':
zvals = likelihood_stats.loglr + likelihood_stats.prior + fp.lognl
zlbl = r'$\log[p(d|\vec{\vartheta})p(\vec{\vartheta})]$'
elif arg == 'prior':
zvals = likelihood_stats.prior
zlbl = r'$\log p(\vec{\vartheta})$'
return zvals, zlbl
def get_zvalues(fp, arg, likelihood_stats):
"""Reads the data for the z-value of the plots from the inference file.
Parameters
----------
fp : InferenceFile
An open inference file; needed to get the value of the log noise
likelihood.
arg : str
The argument to plot; must be one of `loglr`, `snr`, `logplr`,
`logposterior`, or `prior`. If not one of these, a ValueError is
raised.
likelihood_stats : FieldArray
The likelihood stats; the sort of thing returned by
`fp.read_likelihood_stats`.
Returns
-------
zvals : numpy.array
An array of the desired likelihood values to plot.
zlbl : str
The label to use for the values on a plot.
"""
if arg == 'loglr':
zvals = likelihood_stats.loglr
zlbl = r'$\log\mathcal{L}(\vec{\vartheta})$'
elif arg == 'snr':
zvals = likelihood.snr_from_loglr(likelihood_stats.loglr)
zlbl = r'$\rho(\vec{\vartheta})$'
elif arg == 'logplr':
zvals = likelihood_stats.loglr + likelihood_stats.prior
zlbl = r'$\log[\mathcal{L}(\vec{\vartheta})p(\vec{\vartheta})]$'
elif arg == 'logposterior':
zvals = likelihood_stats.loglr + likelihood_stats.prior + fp.lognl
zlbl = r'$\log[p(d|\vec{\vartheta})p(\vec{\vartheta})]$'
elif arg == 'prior':
zvals = likelihood_stats.prior
zlbl = r'$\log p(\vec{\vartheta})$'
else:
raise ValueError("Unrecognized arg {}".format(arg))
return zvals, zlbl
def get_zvalues(fp, arg, likelihood_stats):
"""Reads the data for the z-value of the plots from the inference file.
Parameters
----------
fp : InferenceFile
An open inference file; needed to get the value of the log noise
likelihood.
arg : str
The argument to plot; must be one of `loglr`, `snr`, `logplr`,
`logposterior`, or `prior`. If not one of these, a ValueError is
raised.
likelihood_stats : FieldArray
The likelihood stats; the sort of thing returned by
`fp.read_likelihood_stats`.
Returns
-------
zvals : numpy.array
An array of the desired likelihood values to plot.
zlbl : str
The label to use for the values on a plot.
"""
if arg == 'loglr':
zvals = likelihood_stats.loglr
zlbl = r'$\log\mathcal{L}(\vec{\vartheta})$'
elif arg == 'snr':
zvals = likelihood.snr_from_loglr(likelihood_stats.loglr)
zlbl = r'$\rho(\vec{\vartheta})$'
elif arg == 'logplr':
zvals = likelihood_stats.loglr + likelihood_stats.prior
zlbl = r'$\log[\mathcal{L}(\vec{\vartheta})p(\vec{\vartheta})]$'
elif arg == 'logposterior':
zvals = likelihood_stats.loglr + likelihood_stats.prior + fp.lognl
zlbl = r'$\log[p(d|\vec{\vartheta})p(\vec{\vartheta})]$'
elif arg == 'prior':
zvals = likelihood_stats.prior
zlbl = r'$\log p(\vec{\vartheta})$'
else:
raise ValueError("Unrecognized arg {}".format(arg))
return zvals, zlbl