"GedPhoton":"genEnergy/scCalibratedEnergy", } specificVariables = { "EG":["scRawEnergy"], #"PFBox":["scCalibratedEnergy"], "PFMustache":["scCalibratedEnergy"], "GedPhoton":["scCalibratedEnergy"], } clusterings = [ #("PFBox","regularBoxSCTree/SuperClusterTree"), ("PFMustache","mustacheSCTree/SuperClusterTree"), ("EG","egSCTree/SuperClusterTree"), ("GedPhoton","gedPhotonTree/RegressionTree"), ] for tagClustering,treeClustering in clusterings: # define regressions batch.addRegression(tagClustering) batch.setInputTree(tagClustering, treeClustering) batch.setTarget(tagClustering, targets[tagClustering]) batch.addVariablesEB(tagClustering, specificVariables[tagClustering]) batch.addVariablesEE(tagClustering, specificVariables[tagClustering]) #batch.simulate = True batch.info = info
"EG": "genEnergy/(scRawEnergy+scPreshowerEnergy)", #"PFBox":"genEnergy/scCalibratedEnergy", "PFMustache": "genEnergy/scCalibratedEnergy", "GedPhoton": "genEnergy/scCalibratedEnergy", } specificVariables = { "EG": ["scRawEnergy"], #"PFBox":["scCalibratedEnergy"], "PFMustache": ["scCalibratedEnergy"], "GedPhoton": ["scCalibratedEnergy"], } clusterings = [ #("PFBox","regularBoxSCTree/SuperClusterTree"), ("PFMustache", "mustacheSCTree/SuperClusterTree"), ("EG", "egSCTree/SuperClusterTree"), ("GedPhoton", "gedPhotonTree/RegressionTree"), ] for tagClustering, treeClustering in clusterings: # define regressions batch.addRegression(tagClustering) batch.setInputTree(tagClustering, treeClustering) batch.setTarget(tagClustering, targets[tagClustering]) batch.addVariablesEB(tagClustering, specificVariables[tagClustering]) batch.addVariablesEE(tagClustering, specificVariables[tagClustering]) #batch.simulate = True batch.info = info