diff --git a/FTagDumper/python/blocks/MultifoldTagger.py b/FTagDumper/python/blocks/MultifoldTagger.py index 6b3cb135577e41a8cb0a7f32e3fcf66f0ec2b081..ab182df559afa2e79df6caa9ded9b23a16d2fdb4 100644 --- a/FTagDumper/python/blocks/MultifoldTagger.py +++ b/FTagDumper/python/blocks/MultifoldTagger.py @@ -31,6 +31,8 @@ class MultifoldTagger(BaseBlock): Name of the constituent container. default_zero_tracks : bool Bool to decide how to work with zero tracks. By default False. + flip_tag_config : str + String for if to eval the flipped tagger version (Default STANDARD) """ nn_paths: list[str] alg_name: str = None @@ -40,6 +42,7 @@ class MultifoldTagger(BaseBlock): fold_hash_name: str = "jetFoldHash" constituents: str = "InDetTrackParticles" default_zero_tracks: bool = False + flip_tag_config : str = "STANDARD" def __post_init__(self): @@ -80,6 +83,7 @@ class MultifoldTagger(BaseBlock): variableRemapping=self.remap, trackLinkType=self.track_link_type, defaultZeroTracks=self.default_zero_tracks, + flipTagConfig=self.flip_tag_config ) ) ) diff --git a/FTagDumper/test/test-dumper b/FTagDumper/test/test-dumper index 8220c6bcc794e038a9df3b93dace1ddeda54e31b..b831df613706ae8514f7516f1cefd86f2e910227 100755 --- a/FTagDumper/test/test-dumper +++ b/FTagDumper/test/test-dumper @@ -60,6 +60,7 @@ declare -A CONFIGS=( [blocks]=${CFG_DIR}/BlockDemo.json [smeared-tracks]=${CFG_DIR}/trigger_upgrade.json [gn3]=${CFG_DIR}/GN3_dev.json + [gn2flip]=${CFG_DIR}/GN2SimpleFlip.json [displaced-jets]=${CFG_DIR}/DisplacedJets.json [upgrade-hi]=${CFG_DIR}/upgrade-HI.json [ndive]=${CFG_DIR}/NDIVE.json @@ -128,6 +129,7 @@ declare -A DATAFILES=( [blocks]=${OFFLINE_FTAG1} [smeared-tracks]=p5799/DAOD_FTAG1.601229.e8481_s4149_r14700_r14702_p5799.small.pool.root [gn3]=${OFFLINE_FTAG1} + [gn2flip]=${OFFLINE_FTAG1} [displaced-jets]=${LLP1} [upgrade-hi]=r11199/AOD.800897.e8366_s3238_r11199.small.pool.root [ndive]=${OFFLINE_FTAG1} @@ -184,6 +186,7 @@ declare -A TESTS=( [blocks]=dump-single-btag [smeared-tracks]=dump-smeared-tracks [gn3]=dump-single-btag + [gn2flip]=dump-single-btag [displaced-jets]=dump-displaced-jets [upgrade-hi]=dump-single-btag [ndive]=dump-single-btag diff --git a/configs/GN2SimpleFlip.json b/configs/GN2SimpleFlip.json new file mode 100644 index 0000000000000000000000000000000000000000..976b4b4d988d3055a5d37b2230442d10a68c38fe --- /dev/null +++ b/configs/GN2SimpleFlip.json @@ -0,0 +1,41 @@ +{ + "file": "GN3_dev.json", + "btagging_link": "btaggingLink", + "variables": { + "file": "fragments/pflow-sv1-jf.json", + "btagging": { + "floats": [ + "GN2Lep_pb", + "GN2Lep_pc", + "GN2Lep_pu", + "GN2Lep_ptau", + "GN2LepSimpleFlip_pb", + "GN2LepSimpleFlip_pc", + "GN2LepSimpleFlip_pu", + "GN2LepSimpleFlip_ptau" + ] + } + }, + "ca_blocks": [ + { + "block": "MultifoldTagger", + "nn_paths": [ + "dev/BTagging/20240627/GN2v01Lep/antikt4empflow/network_fold0.onnx", + "dev/BTagging/20240627/GN2v01Lep/antikt4empflow/network_fold1.onnx", + "dev/BTagging/20240627/GN2v01Lep/antikt4empflow/network_fold2.onnx", + "dev/BTagging/20240627/GN2v01Lep/antikt4empflow/network_fold3.onnx" + ] + }, + { + "block": "MultifoldTagger", + "flip_tag_config": "SIMPLE_FLIP", + "alg_name": "flip_tagger", + "nn_paths": [ + "dev/BTagging/20240627/GN2v01Lep/antikt4empflow/network_fold0.onnx", + "dev/BTagging/20240627/GN2v01Lep/antikt4empflow/network_fold1.onnx", + "dev/BTagging/20240627/GN2v01Lep/antikt4empflow/network_fold2.onnx", + "dev/BTagging/20240627/GN2v01Lep/antikt4empflow/network_fold3.onnx" + ] + } + ] +} diff --git a/configs/fragments/pflow-sv1-jf.json b/configs/fragments/pflow-sv1-jf.json new file mode 100644 index 0000000000000000000000000000000000000000..8e198251af59131e62ca881590378c6da2bfff07 --- /dev/null +++ b/configs/fragments/pflow-sv1-jf.json @@ -0,0 +1,46 @@ +{ + "btagging" : { + "ints" : [ + "JetFitter_nVTX", + "JetFitter_nSingleTracks", + "JetFitter_nTracksAtVtx", + "JetFitter_N2Tpair", + "SV1_N2Tpair", + "SV1_NGTinSvx", + "JetFitterSecondaryVertex_nTracks" + ], + "floats": [ + "JetFitter_energyFraction", + "JetFitter_mass", + "JetFitter_significance3d", + "JetFitter_deltaphi", + "JetFitter_deltaeta", + "JetFitter_massUncorr", + "JetFitter_dRFlightDir", + "SV1_masssvx", + "SV1_efracsvx", + "SV1_significance3d", + "SV1_dstToMatLay", + "SV1_deltaR", + "SV1_Lxy", + "SV1_L3d", + "JetFitter_deltaR", + "JetFitterSecondaryVertex_displacement3d", + "JetFitterSecondaryVertex_displacement2d", + "JetFitterSecondaryVertex_mass", + "JetFitterSecondaryVertex_energy", + "JetFitterSecondaryVertex_energyFraction", + "JetFitterSecondaryVertex_minimumTrackRelativeEta", + "JetFitterSecondaryVertex_maximumTrackRelativeEta", + "JetFitterSecondaryVertex_averageTrackRelativeEta", + "JetFitterSecondaryVertex_maximumAllJetTrackRelativeEta", + "JetFitterSecondaryVertex_minimumAllJetTrackRelativeEta", + "JetFitterSecondaryVertex_averageAllJetTrackRelativeEta" + ], + "chars": [ + "JetFitter_isDefaults", + "SV1_isDefaults", + "JetFitterSecondaryVertex_isDefaults" + ] + } +}