예제 #1
0
    def applyMVA(self, name, SelB, MVAVars, MVAxmlFile, MVACutValue):
        from MVADictHelpers import addTMVAclassifierValue
        from Configurables import FilterDesktop as MVAFilterDesktop

        _FilterB = MVAFilterDesktop(name + "Filter",
                                    Code="VALUE('LoKi::Hybrid::DictValue/" +
                                    name + "')>" + MVACutValue)

        addTMVAclassifierValue(Component=_FilterB,
                               XMLFile=MVAxmlFile,
                               Variables=MVAVars,
                               ToolName=name)

        return Selection(name, Algorithm=_FilterB, RequiredSelections=[SelB])
예제 #2
0
def applyBDT(name, SelB2HHBDT, BDTVars, BDTCutValue, BDTWeightsFile):

    from MVADictHelpers import addTMVAclassifierValue
    from Configurables import FilterDesktop as MVAFilterDesktop

    _FilterB = MVAFilterDesktop(name + "Filter",
                                Code="VALUE('LoKi::Hybrid::DictValue/" + name +
                                "')>" + str(BDTCutValue))

    addTMVAclassifierValue(Component=_FilterB,
                           XMLFile=BDTWeightsFile,
                           Variables=BDTVars,
                           ToolName=name)

    return Selection(name, Algorithm=_FilterB, RequiredSelections=[SelB2HHBDT])