Esempio n. 1
0
def toolMV2c00Tag(name, useBTagFlagsDefaults=True, **options):
    """Sets up a MV2c00Tag tool and returns it.

    The following options have BTaggingFlags defaults:

    OutputLevel                         default: BTaggingFlags.OutputLevel
    Runmodus                            default: BTaggingFlags.Runmodus
    taggerName                          default: "MV2c00"
    taggerNameBase                      default: "MV2c00"
    forceMV2CalibrationAlias            default: BTaggingFlags.ForceMV2CalibrationAlias
    MV2CalibAlias                       default: BTaggingFlags.MV2CalibAlias

    input:             name: The name of the tool (should be unique).
      useBTagFlagsDefaults : Whether to use BTaggingFlags defaults for options that are not specified.
                  **options: Python dictionary with options for the tool.
    output: The actual tool, which can then by added to ToolSvc via ToolSvc += output."""
    from BTagging.MV2defaultValues import default_values
    from BTagging.MV2defaultValues import MVTM_varNames

    if useBTagFlagsDefaults:
        defaults = {
            'OutputLevel': BTaggingFlags.OutputLevel,
            'Runmodus': BTaggingFlags.Runmodus,
            'taggerName': 'MV2c00',
            'taggerNameBase': 'MV2c00',
            'forceMV2CalibrationAlias': BTaggingFlags.ForceMV2CalibrationAlias,
            'MV2CalibAlias': BTaggingFlags.MV2CalibAlias,
            'defaultvals': default_values,
            'MVTMvariableNames': MVTM_varNames,
        }
        for option in defaults:
            options.setdefault(option, defaults[option])
    options['name'] = name
    from JetTagTools.JetTagToolsConf import Analysis__MV2Tag
    return Analysis__MV2Tag(**options)
Esempio n. 2
0
def toolMV2Tag(name, useBTagFlagsDefaults=True, **options):
    """Sets up a MV2Tag tool and returns it.

    The following options have BTaggingFlags defaults:

    OutputLevel                         default: BTaggingFlags.OutputLevel
    Runmodus                            default: BTaggingFlags.Runmodus
    taggerName                          default: "MV2Tag"
    taggerNameBase                      default: "MV2Tag"

    input:             name: The name of the tool (should be unique).
      useBTagFlagsDefaults : Whether to use BTaggingFlags defaults for options that are not specified.
                  **options: Python dictionary with options for the tool.
    output: The actual tool, which can then by added to ToolSvc via ToolSvc += output."""
    if useBTagFlagsDefaults:
        defaults = {
            'OutputLevel': BTaggingFlags.OutputLevel,
            'Runmodus': BTaggingFlags.Runmodus,
            'taggerName': 'MV2Tag',
            'taggerNameBase': 'MV2Tag'
        }
        for option in defaults:
            options.setdefault(option, defaults[option])
    options['name'] = name
    from JetTagTools.JetTagToolsConf import Analysis__MV2Tag
    return Analysis__MV2Tag(**options)
Esempio n. 3
0
def toolMV2c20Tag(name, useBTagFlagsDefaults=True, **options):
    """Sets up a MV2c20Tag tool and returns it.

    The following options have BTaggingFlags defaults:

    OutputLevel                         default: BTaggingFlags.OutputLevel
    Runmodus                            default: BTaggingFlags.Runmodus
    DecorateMvaInputs                   default: BTaggingFlags.DecorateMvaInputs
    taggerName                          default: "MV2c20"
    taggerNameBase                      default: "MV2c20"
    forceMV2CalibrationAlias            default: BTaggingFlags.ForceMV2CalibrationAlias
    MV2CalibAlias                       default: BTaggingFlags.MV2CalibAlias
    inputSV0SourceName                  default: "SV0"
    inputSV1SourceName                  default: "SV1"
    inputIP2DSourceName                 default: "IP2D"
    inputIP3DSourceName                 default: "IP3D"
    inputJFSourceName                   default: "JetFitter"
    inputJFProbSourceName               default: "JetFitterCombNN"
    trainingConfig                      default: BTaggingFlags.MV2cTrainingConfig

    input:             name: The name of the tool (should be unique).
      useBTagFlagsDefaults : Whether to use BTaggingFlags defaults for options that are not specified.
                  **options: Python dictionary with options for the tool.
    output: The actual tool, which can then by added to ToolSvc via ToolSvc += output."""
    if useBTagFlagsDefaults:
        defaults = {
            'OutputLevel': BTaggingFlags.OutputLevel,
            'Runmodus': BTaggingFlags.Runmodus,
            'DecorateMvaInputs': BTaggingFlags.DecorateMvaInputs,
            'taggerName': 'MV2c20',
            'taggerNameBase': 'MV2c20',
            'forceMV2CalibrationAlias': BTaggingFlags.ForceMV2CalibrationAlias,
            'MV2CalibAlias': BTaggingFlags.MV2CalibAlias,
            'inputSV0SourceName': 'SV0',
            'inputSV1SourceName': 'SV1',
            'inputIP2DSourceName': 'IP2D',
            'inputIP3DSourceName': 'IP3D',
            'inputJFSourceName': 'JetFitter',
            'inputJFProbSourceName': 'JetFitterCombNN',
            'trainingConfig': BTaggingFlags.MV2cTrainingConfig,
        }
        for option in defaults:
            options.setdefault(option, defaults[option])
    options['name'] = name
    from JetTagTools.JetTagToolsConf import Analysis__MV2Tag
    return Analysis__MV2Tag(**options)
from JetTagTools.JetTagToolsConf import Analysis__MV2Tag
MV2c00FlipTagTool = Analysis__MV2Tag(
    name="MV2c00FlipTag",
    Runmodus=BTaggingFlags.Runmodus,
    calibrationTool=BTagCalibrationBrokerTool,
    forceMV2CalibrationAlias=BTaggingFlags.ForceMV2CalibrationAlias,
    MV2CalibAlias=BTaggingFlags.MV2CalibAlias,
    OutputLevel=BTaggingFlags.OutputLevel,
    flipMV2=True)
ToolSvc += MV2c00FlipTagTool
if BTaggingFlags.OutputLevel < 3:
    print MV2c00FlipTagTool
from JetTagTools.JetTagToolsConf import Analysis__MV2Tag
MV2c00TagTool = Analysis__MV2Tag(
                              name = "MV2c00Tag",
                              Runmodus = BTaggingFlags.Runmodus,
                              calibrationTool = BTagCalibrationBrokerTool,
                              forceMV2CalibrationAlias = BTaggingFlags.ForceMV2CalibrationAlias,
                              MV2CalibAlias =  BTaggingFlags.MV2CalibAlias,
                              xAODBaseName="MV2c00",
                              inputSV0SourceName = "SV0",
                              inputSV1SourceName = "SV1",
                              inputIP2DSourceName = "IP2D",
                              inputIP3DSourceName = "IP3D",
                              inputJFSourceName = "JetFitter",
                              inputJFProbSourceName = "JetFitterCombNN",
                              OutputLevel = BTaggingFlags.OutputLevel
                             )
ToolSvc += MV2c00TagTool
if BTaggingFlags.OutputLevel < 3:
  print MV2c00TagTool