Skip to content

Instantly share code, notes, and snippets.

@kratsg
Created October 28, 2020 20:54
Show Gist options
  • Save kratsg/03dd20fc3a1ac8c057ff67b7d831e03f to your computer and use it in GitHub Desktop.
Save kratsg/03dd20fc3a1ac8c057ff67b7d831e03f to your computer and use it in GitHub Desktop.
import pyhf
pyhf.set_backend('pytorch', 'minuit')
nSig = 4.166929245
errSig = 4.166929245
nBkg = 0.11
errBkgUp = 0.20
errBkgDown = 0.11
model_json = {
"channels": [
{ "name": "SR_combined",
"samples": [
{ "name": "signal",
"data": [nSig],
"modifiers": [ { "name": "mu", "type": "normfactor", "data": None},
# {"name": "uncorr_siguncrt", "type": "shapesys", "data": [0.1]}
]
},
{ "name": "background",
"data": [nBkg],
"modifiers": [ {"name": "uncorr_bkguncrt", "type": "normsys", "data": {"hi": errBkgUp, "lo": errBkgDown}} ]
}
]
}
],
"observations": [
{ "name": "SR_combined", "data": [0.0] }
],
"measurements": [
{ "name": "Measurement", "config": {"poi": "mu", "parameters": []} }
],
"version": "1.0.0"
}
ws = pyhf.Workspace(model_json)
model = ws.model()
data = ws.data(model)
pyhf.infer.mle.fit(data, model) # ok
pyhf.infer.hypotest(1.0, data, model, qtilde=True) # ok
pyhf.infer.hypotest(1.0, data, model, qtilde=True, calctype='toybased', ntoys=100)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment