Python RooWjjMjjFitter.computeChi2示例

    # theFitter.getWorkSpace().var('nSingleTop').setConstant()
    theFitter.getWorkSpace().var('nQCD').setConstant()
    # theFitter.getWorkSpace().var('nZjets').setConstant()
    #theFitter.getWorkSpace().var('nWjets').setConstant()
    fr = theFitter.fit()
    ndf = Long(fr.floatParsFinal().getSize())
    tries += 1

mass = theFitter.getWorkSpace().var(fitterPars.var)
iset = RooArgSet(mass)
nexp = theFitter.makeFitter().expectedEvents(iset)

ndf2 = Long(ndf)
print 'Corrected chi2'
#chi2 = ndf*1.0
chi2 = theFitter.computeChi2(ndf)
print 'Raw chi2'
#chi2Raw = ndf2*1.0
chi2Raw = theFitter.computeChi2(ndf2, False)

mf = theFitter.stackedPlot()
mf.SetName("Mjj_Stacked")
## sf = theFitter.residualPlot(mf, "h_background", "dibosonPdf", False)
# pf = theFitter.residualPlot(mf, "h_total", "", True)
pf = pulls.createPull(mf.getHist('theData'), mf.getCurve('h_total'))
pf.SetName("Mjj_Pull")
## lf = theFitter.stackedPlot(True)

l = TLatex()
l.SetNDC()
l.SetTextSize(0.035)

示例#2

显示文件

文件： runDibosonFitter.py 项目： cms-analysis/ElectroWeakAnalysis-VPlusJets

#theFitter.getWorkSpace().Print()
fr = theFitter.fit()

# chi2 = Double(0.)
#ndf = Long(2)
extraNdf = 0
if not opts.btag:
    extraNdf = 2
## if (fitterPars.doNewPhysics):
##     extraNdf += 1
## if not fitterPars.constrainDiboson:
##     extraNdf += 1
## ndf = Long(3+extraNdf)
ndf = Long(1+extraNdf)
print ' *** uncorrected chi2 ***'
chi2 = theFitter.computeChi2(ndf, False)
print ' *** corrected chi2 ***'
ndf = Long(1+extraNdf)
chi2 = theFitter.computeChi2(ndf, True)
print ' *** '
chi2 = 0
# chi2frame.Draw()

# assert False, "fit done"

yields = fr.floatParsFinal()
mass = theFitter.getWorkSpace().var(fitterPars.var)
iset = RooArgSet(mass)
covMatrix = TMatrixDSym(fr.covarianceMatrix())

sig2 = 0.

示例#3

显示文件

文件： runHWWFitter.py 项目： VPlusJetsAnalyzers/VPlusJets

    # theFitter.getWorkSpace().var('nSingleTop').setConstant()
    theFitter.getWorkSpace().var('nQCD').setConstant()
    # theFitter.getWorkSpace().var('nZjets').setConstant()
    #theFitter.getWorkSpace().var('nWjets').setConstant()
    fr = theFitter.fit()
    ndf = Long(fr.floatParsFinal().getSize())
    tries += 1

mass = theFitter.getWorkSpace().var(fitterPars.var)
iset = RooArgSet(mass)
nexp = theFitter.makeFitter().expectedEvents(iset)

ndf2 = Long(ndf)
print 'Corrected chi2'
#chi2 = ndf*1.0
chi2 = theFitter.computeChi2(ndf)
print 'Raw chi2'
#chi2Raw = ndf2*1.0
chi2Raw = theFitter.computeChi2(ndf2, False)

mf = theFitter.stackedPlot()
mf.SetName("Mjj_Stacked")
## sf = theFitter.residualPlot(mf, "h_background", "dibosonPdf", False)
# pf = theFitter.residualPlot(mf, "h_total", "", True)
pf = pulls.createPull(mf.getHist('theData'), mf.getCurve('h_total'))
pf.SetName("Mjj_Pull")
## lf = theFitter.stackedPlot(True)

l = TLatex()
l.SetNDC()
l.SetTextSize(0.035)

示例#4

显示文件

文件： runMjjFitter.py 项目： cms-analysis/ElectroWeakAnalysis-VPlusJets

    modeString = 'Electron'
else:
    modeString = ''

theFitter = RooWjjMjjFitter(fitterPars)

theFitter.makeFitter(False)

#theFitter.getWorkSpace().Print()
fr = theFitter.fit()

extraNdf = 0
if not fitterPars.constrainDiboson:
    extraNdf += 1
ndf = Long(fr.floatParsFinal().getSize()-5+extraNdf)
chi2 = theFitter.computeChi2(ndf)
# chi2frame.Draw()

# assert False, "fit done"

yields = fr.floatParsFinal()
mass = theFitter.getWorkSpace().var(fitterPars.var)
iset = RooArgSet(mass)
covMatrix = TMatrixDSym(fr.covarianceMatrix())

sig2 = 0.
for v1 in range(0, covMatrix.GetNrows()):
    for v2 in range(0, covMatrix.GetNcols()):
        if ((yields[v1].GetName())[0] == 'n') and \
               ((yields[v2].GetName())[0] == 'n'):
            #print v1, yields[v1].GetName(),',', v2, yields[v2].GetName()