Skip to content
Snippets Groups Projects

Run ML training on 100 events for every pipeline (manual for 100k)

Merged Robinpreet Dhillon requested to merge robin-ML_test into master
Compare and
6 files
+ 332
294
Compare changes
  • Side-by-side
  • Inline
Files
6
@@ -15,8 +15,6 @@
#include "eicd/CalorimeterHitData.h"
#include "eicd/ClusterCollection.h"
#include "eicd/ClusterData.h"
#include "eicd/ClusterLayerData.h"
#include "eicd/RawCalorimeterHitCollection.h"
#include "eicd/RawCalorimeterHitData.h"
@@ -229,7 +227,7 @@ void emcal_barrel_pion_rejection_analysis(
// EcalBarrelImagingClustersLayers Functions////////////////////////////////////////////////////////////////////////////////
// Number of hits
auto nhitsClusLayerImg = [] (const std::vector<eicd::ClusterLayerData>& evt) {
auto nhitsClusLayerImg = [] (const std::vector<eicd::ClusterData>& evt) {
int nhitsTot = 0;
for (const auto &i : evt){
nhitsTot += i.nhits;
@@ -238,10 +236,10 @@ void emcal_barrel_pion_rejection_analysis(
};
// Number of clusters
auto nClusLayerImg = [] (const std::vector<eicd::ClusterLayerData>& evt) {return (int) evt.size(); };
auto nClusLayerImg = [] (const std::vector<eicd::ClusterData>& evt) {return (int) evt.size(); };
// Energy deposition in cluster [GeV]
auto EClusLayerImg = [](const std::vector<eicd::ClusterLayerData>& evt) {
auto EClusLayerImg = [](const std::vector<eicd::ClusterData>& evt) {
double total_edep = 0.0;
for (const auto& i: evt){
total_edep += i.energy;
@@ -250,7 +248,7 @@ void emcal_barrel_pion_rejection_analysis(
};
// Max Energy deposition in cluster [GeV]
auto EClusLayerMaxImg = [](const std::vector<eicd::ClusterLayerData>& evt) {
auto EClusLayerMaxImg = [](const std::vector<eicd::ClusterData>& evt) {
double max = 0.0;
for (const auto& i: evt){
if (i.energy > max){max = i.energy;}
@@ -259,7 +257,7 @@ void emcal_barrel_pion_rejection_analysis(
};
// Min Energy deposition in cluster [GeV]
auto EClusLayerMax2Img = [](const std::vector<eicd::ClusterLayerData>& evt) {
auto EClusLayerMax2Img = [](const std::vector<eicd::ClusterData>& evt) {
double max1 = 0.0;
double max2 = 0.0;
for (const auto& i: evt){
Loading