Skip to content
Snippets Groups Projects

Add optional ML based e/p workflow

Merged Mark Hodgkinson requested to merge atlas-particleflow-software/athena:pflow_NN into 23.0
1 file
+ 0
16
Compare changes
  • Side-by-side
  • Inline
@@ -150,11 +150,6 @@ float PFEnergyPredictorTool::nnEnergyPrediction(const eflowRecTrack *ptr) const{
std::vector<eflowRecCluster*> matchedClusters;
std::vector<eflowTrackClusterLink*> links = ptr->getClusterMatches();
std::stringstream phi;
std::stringstream eta;
std::stringstream clustphi;
std::stringstream clusteta;
std::array<double, 19> etatotal = getEtaTrackCalo(ptr->getTrackCaloPoints());
std::array<double, 19> phitotal = getPhiTrackCalo(ptr->getTrackCaloPoints());
@@ -183,15 +178,8 @@ float PFEnergyPredictorTool::nnEnergyPrediction(const eflowRecTrack *ptr) const{
totalE += clusterE;
matchedClusters.push_back(clink->getCluster());
phi << clink->getCluster()->getCluster()->phi();
phi << ',';
eta << clink->getCluster()->getCluster()->eta();
eta << ',';
}
phi << links.size();
eta << links.size();
std::vector<std::array<double, 5>> cells;
const eflowTrackCaloPoints& trackCaloPoints = ptr->getTrackCaloPoints();
@@ -228,10 +216,6 @@ float PFEnergyPredictorTool::nnEnergyPrediction(const eflowRecTrack *ptr) const{
auto theDDE=it_cell->caloDDE();
double cx=theDDE->x();
double cy=theDDE->y();
clustphi << theDDE->phi();
clustphi << ',';
clusteta << theDDE->eta();
clusteta << ',';
cells.emplace_back( std::array<double, 5> { cellE,
theDDE->eta() - eta_ctr,
Loading