3 #include "CAFAna/Core/Var.h" 9 #include "TensorFlowEvaluator/SliceLID/cafana/ResultCacher.hxx" 19 const std::vector<std::pair<std::string, Var>> &sliceVarDefs,
20 const std::vector<std::pair<std::string, MultiVar>> &png2dVarDefs,
21 const std::vector<std::pair<std::string, MultiVar>> &png3dVarDefs
26 for (
auto &namedVar : sliceVarDefs) {
27 result.
scalarVarMap[namedVar.first] = namedVar.second(sr);
30 for (
auto &namedMultiVar : png2dVarDefs) {
31 result.
vectorVarMap[namedMultiVar.first] = namedMultiVar.second(sr);
34 for (
auto &namedMultiVar : png3dVarDefs) {
35 result.
vectorVarMap[namedMultiVar.first] = namedMultiVar.second(sr);
45 auto model = std::make_shared<Model>(modelSavedir);
47 return std::make_shared<CAFAnaModel>(
49 model = std::move(model),
60 trkLenCorr.shift(varDict);
62 return model->predict(varDict);
72 return (*model)(sr).primaryE;
82 return (*model)(sr).totalE;
92 return (*model)(sr).totalE - (*model)(sr).primaryE;
Far Detector at Ash River.
Cuts and Vars for the 2020 FD DiF Study.
SRHeader hdr
Header branch: run, subrun, etc.
std::unordered_map< std::string, std::vector< double > > vectorVarMap
Proxy for caf::StandardRecord.
const std::vector< std::pair< std::string, MultiVar > > kPng2dVarDefs
static Var secondaryEnergy(const std::shared_ptr< CAFAnaModel > &model)
_Var< caf::SRProxy > Var
Representation of a variable to be retrieved from a caf::StandardRecord object.
const std::vector< std::pair< std::string, Var > > kSliceVarDefs
short nallcosmics
Number of cosmics in allcosmics vector.
static Var primaryEnergy(const std::shared_ptr< CAFAnaModel > &model)
const std::vector< std::pair< std::string, MultiVar > > kPng3dVarDefs
static std::shared_ptr< CAFAnaModel > initCAFAnaModel(const std::string &modelSavedir)
ResultCacher< LSTMEnergy > CAFAnaModel
static VarDict fillVarDict(const caf::SRProxy *sr, const std::vector< std::pair< std::string, Var >> &sliceVarDefs, const std::vector< std::pair< std::string, MultiVar >> &png2dVarDefs, const std::vector< std::pair< std::string, MultiVar >> &png3dVarDefs)
SRTruthBranch mc
Truth branch for MC: energy, flavor, etc.
static Var totalEnergy(const std::shared_ptr< CAFAnaModel > &model)
short nallnus
Number of neutrinos in allnus vector.
const XML_Char XML_Content * model
std::unordered_map< std::string, double > scalarVarMap