1 #ifndef ALITRDPIDREFMAKERNN_H
2 #define ALITRDPIDREFMAKERNN_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
6 /* $Id: AliTRDpidRefMakerNN.h 27496 2008-07-22 08:35:45Z cblume $ */
8 //////////////////////////////////////////////////////
10 // Task to build PID reference tree for the training
11 // of neural networs for the TRD PID
13 // Authors: Alex Wilk <wilka@uni-muenster.de>
14 // Markus Heide <mheide@uni-muenster.de>
16 ///////////////////////////////////////////////////////
18 #ifndef ALITRDPIDREFMAKER_H
19 #include "AliTRDpidRefMaker.h"
21 #ifndef ALITRDCALPID_H
22 #include "AliTRDCalPID.h"
26 class TMultiLayerPerceptron;
27 class AliTRDpidRefMakerNN : public AliTRDpidRefMaker
31 enum ETRDpidRefMakerNNgraph {
36 enum ETRDpidRefMakerNNmoni {
40 enum ETRDpidRefMakerNNsteer{
41 kMaxStat = 20000 // maximum statistics/PID bin
42 ,kMinStat = 50 // minimum statistics/PID bin
44 AliTRDpidRefMakerNN();
45 AliTRDpidRefMakerNN(const char *n);
46 virtual ~AliTRDpidRefMakerNN();
48 Int_t GetEpochs() const {return fEpochs;};
49 Int_t GetMinTrain() const {return fMinTrain;};
50 Int_t GetTrainMomBin() const {return fTrainMomBin;};
54 void SetEpochs(Int_t epochs) {fEpochs = epochs;};
55 void SetMinTrain(Int_t mintrain) {fMinTrain = mintrain;};
56 void SetTrainMomBin(Int_t trainmombin) {fTrainMomBin = trainmombin;};
57 void SetDate(Int_t date) {fDate = date;};
58 void SetDoTraining(Bool_t train) {fDoTraining = train;};
59 void SetContinueTraining(Bool_t continTrain) {fContinueTraining = continTrain;};
60 void SetTrainPath(Int_t path) {fTrainPath = path;};
61 Bool_t LoadFile(const Char_t *InFileNN);
62 void SetScaledEdx(Float_t s) {fScale = s;};
64 Bool_t MakeTrainingSample(); // convert AnalysisResults.root to training file
65 void MakeTrainingLists(Int_t mombin = 0); // build the training and the test list
66 void MonitorTraining(Int_t mombin); // monitor training process
69 void MakeRefs(Int_t mombin); // train the neural networks for a given momentum bin
72 AliTRDpidRefMakerNN(const AliTRDpidRefMakerNN&); // not implemented
73 AliTRDpidRefMakerNN& operator=(const AliTRDpidRefMakerNN&); // not implemented
74 void MakeTrainTestTrees();
76 TEventList *fTrain[AliTRDCalPID::kNMom]; // Training list for each momentum
77 TEventList *fTest[AliTRDCalPID::kNMom]; // Test list for each momentum
78 TMultiLayerPerceptron *fNet; // artificial neural network
80 Int_t fTrainMomBin; // momentum bin for the training
81 Int_t fEpochs; // Number of epochs for the training of the NNs
82 Int_t fMinTrain; // minimum of events needed for training
83 Int_t fDate; // date stamp for training of the NNs
84 Bool_t fDoTraining; // checks if training will be done
85 Bool_t fContinueTraining; // checks if training from an older run should be continued
86 Int_t fTrainPath; // sets the path for continuing the training
88 Float_t fScale; // scaling factor
89 Int_t fLy; // TRD layer
90 Int_t fNtrkl; // No. tracklets
91 TTree *fTrainData[AliTRDCalPID::kNMom];//Tree for reference data for all momentum bins
92 TFile *fRef; //file containing reference data
94 ClassDef(AliTRDpidRefMakerNN, 3); // TRD reference maker for NN