Logo ROOT   6.30.04
Reference Guide
 All Namespaces Files Pages
SVWorkingSet.h
Go to the documentation of this file.
1 // @(#)root/tmva $Id$
2 // Author: Andrzej Zemla
3 
4 /**********************************************************************************
5  * Project: TMVA - a Root-integrated toolkit for multivariate data analysis *
6  * Package: TMVA *
7  * Class : SVWorkingSet *
8  * Web : http://tmva.sourceforge.net *
9  * *
10  * Description: *
11  * Working class for Support Vector Machine *
12  * *
13  * Authors (alphabetical): *
14  * Marcin Wolter <Marcin.Wolter@cern.ch> - IFJ PAN, Krakow, Poland *
15  * Andrzej Zemla <azemla@cern.ch> - IFJ PAN, Krakow, Poland *
16  * (IFJ PAN: Henryk Niewodniczanski Inst. Nucl. Physics, Krakow, Poland) *
17  * *
18  * Copyright (c) 2005: *
19  * CERN, Switzerland *
20  * MPI-K Heidelberg, Germany *
21  * PAN, Krakow, Poland *
22  * *
23  * Redistribution and use in source and binary forms, with or without *
24  * modification, are permitted according to the terms listed in LICENSE *
25  * (http://tmva.sourceforge.net/LICENSE) *
26  **********************************************************************************/
27 
28 #ifndef ROOT_TMVA_SVWorkingSet
29 #define ROOT_TMVA_SVWorkingSet
30 
31 #include <vector>
32 
33 #include "Rtypes.h"
34 
35 namespace TMVA {
36 
37  class SVEvent;
38  class SVKernelMatrix;
39  class SVKernelFunction;
40  class MsgLogger;
41 
42  class SVWorkingSet {
43 
44  public:
45 
46  SVWorkingSet();
47  SVWorkingSet( std::vector<TMVA::SVEvent*>*, SVKernelFunction*, Float_t , Bool_t);
48  ~SVWorkingSet();
49 
50  Bool_t ExamineExample( SVEvent*);
51  Bool_t TakeStep ( SVEvent*, SVEvent*);
52  Bool_t Terminated();
53  void Train(UInt_t nIter=1000);
54  void PrintStat();
55  std::vector<TMVA::SVEvent*>* GetSupportVectors();
56  Float_t GetBpar() {return 0.5*(fB_low + fB_up);}
57 
58  //for regression
59  Bool_t ExamineExampleReg(SVEvent*);
60  Bool_t TakeStepReg(SVEvent*, SVEvent*);
61  Bool_t IsDiffSignificant(Float_t, Float_t, Float_t);
62  void TrainReg();
63 
64  //setting up helper variables for JsMVA
65  void SetIPythonInteractive(bool* ExitFromTraining, UInt_t *fIPyCurrentIter_){
66  fExitFromTraining = ExitFromTraining;
67  fIPyCurrentIter = fIPyCurrentIter_;
68  }
69 
70 
71  private:
72 
73  Bool_t fdoRegression; //TODO temporary, find nicer solution
74  std::vector<TMVA::SVEvent*> *fInputData; // input events
75  std::vector<TMVA::SVEvent*> *fSupVec; // output events - support vectors
76  SVKernelFunction *fKFunction; // kernel function
77  SVKernelMatrix *fKMatrix; // kernel matrix
78 
79  SVEvent *fTEventUp; // last optimized event
80  SVEvent *fTEventLow; // last optimized event
81 
82  Float_t fB_low; // documentation
83  Float_t fB_up; // documentation
84  Float_t fTolerance; // documentation
85 
86  mutable MsgLogger* fLogger; //! message logger
87 
88  // variables for JsMVA
89  UInt_t *fIPyCurrentIter = nullptr;
90  bool * fExitFromTraining = nullptr;
91 
92  void SetIndex( TMVA::SVEvent* );
93  };
94 }
95 
96 #endif //ROOT_TMVA_SVWorkingSet