Import XML¶

In [353]:

%pylab inline

Populating the interactive namespace from numpy and matplotlib

In [366]:

import pyhf
import pyhf.readxml
reload(pyhf.readxml)

spec = pyhf.readxml.parse('../../validation/xmlimport_input/config/example.xml', '../../validation/xmlimport_input/')
pdf = pyhf.Model(spec['channels'])
pdf

Out[366]:

<pyhf.Model at 0x131e068d0>

In [367]:

data = [binvalue for k in pdf.config.channel_order for binvalue in spec['data'][k]] + pdf.auxdata
data

Out[367]:

[122.0, 112.0, 0, 0, 0]

In [368]:

print(pdf.config.suggested_init())
pdf.expected_actualdata(pdf.config.suggested_init())

[0.0, 1.0, 0.0, 0.0]

Out[368]:

[120.0, 110.0]

In [376]:

def assemble(pdf, **kwargs):
    pars = pdf.config.suggested_init()
    for k, v in kwargs.items():
        pars[pdf.config.par_slice(k)] = v
    return pars

def disassemble(pdf, pars):
    return {
        k: pars[pdf.config.par_slice(k)] for k in pdf.config.par_map
    }

print pdf.expected_data(assemble(pdf, SigXsecOverSM = [1.0], syst1 = [0.0], syst2 = [0.0], syst3 = [0.0]))

pars = assemble(pdf, SigXsecOverSM = [2.0], syst2 = [0.0], syst3 = [0.0])
disassemble(pdf,pars)


spec['channels']['channel1']['signal']['data']

[120. 110.   0.   0.   0.]

Out[376]:

array([20., 10.], dtype=float32)

In [370]:

mutests = np.linspace(0,3)
results = [pyhf.runOnePoint(mu, data, pdf, pdf.config.suggested_init(), pdf.config.suggested_bounds()) for mu in mutests]

In [371]:

obs = [r[-2:][0] for r in results]
exp = [r[-2:][1][i] for r in results] for i in range(5)]

def plot_results(testmus,cls_obs, cls_exp, test_size = 0.05):
    plt.plot(testmus,cls_obs, c = 'k')
    for i,c in zip(range(5),['grey','grey','grey','grey','grey']):
        plt.plot(testmus,cls_exp[i], c = c)
    plt.plot(testmus,[test_size]*len(testmus), c = 'r')
    plt.ylim(0,1)
    
plot_results(mutests,obs,exp)