1 #ifndef NeuralNetTrainer_h
2 #define NeuralNetTrainer_h
5 #include "marlin/Processor.h"
7 #include "EVENT/ReconstructedParticle.h"
10 #include "nnet/inc/NeuralNet.h"
11 #include "nnet/inc/NeuralNetDataSet.h"
12 #include "nnet/inc/BackPropagationCGAlgorithm.h"
79 virtual void processRunHeader( LCRunHeader* pRun );
80 virtual void processEvent( LCEvent* pEvent );
86 std::string _JetCollectionName;
87 std::string _FlavourTagInputsCollectionName;
88 std::string _TrueJetFlavourCollectionName;
90 nnet::NeuralNet::SerialisationMode _outputFormat;
97 std::map<std::string,nnet::NeuralNetDataSet*>
_dataSet;
125 #endif //ifndef NeuralNetTrainer_h
std::map< std::string, int > _numBackground
void _trainNet(nnet::BackPropagationCGAlgorithm &pBackPropCGAlgo, nnet::NeuralNetDataSet &dataSet)
Trains neural networks to be used for jet flavour tagging.
std::vector< std::string > _listOfSelectedNetNames
std::map< std::string, std::string > _filename
std::map< std::string, unsigned int > _IndexOf
std::map< std::string, bool > _trainThisNet
std::map< std::string, nnet::NeuralNetDataSet * > _dataSet
std::map< std::string, int > _numSignal
bool _passesCuts(lcio::LCEvent *pEvent)
void _displayCollectionNames(lcio::LCEvent *pEvent)