Logo ROOT   6.30.04
Reference Guide
 All Namespaces Files Pages
SVEvent.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 : SVEvent *
8  * Web : http://tmva.sourceforge.net *
9  * *
10  * Description: *
11  * Event 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_SVEvent
29 #define ROOT_TMVA_SVEvent
30 
31 #include <vector>
32 #include <iostream>
33 
34 #include "Rtypes.h"
35 
36 namespace TMVA
37 {
38  class Event;
39 
40  class SVEvent {
41 
42  public:
43 
44  SVEvent();
45  SVEvent( const Event*, Float_t, Bool_t isSignal = kFALSE );
46  SVEvent( const std::vector<Float_t>*, Float_t alpha, Int_t typeFlag, UInt_t ns );
47  SVEvent( const std::vector<Float_t>* svector, Float_t alpha, Float_t alpha_p, Int_t typeFlag);
48 
49  virtual ~SVEvent();
50 
51  void SetAlpha ( Float_t alpha ) { fAlpha = alpha; }
52  void SetAlpha_p ( Float_t alpha ) { fAlpha_p = alpha; }
53  void SetErrorCache ( Float_t err_cache ) { fErrorCache = err_cache; }
54  void SetIsShrinked ( Int_t isshrinked ) { fIsShrinked = isshrinked; }
55  void SetLine ( Float_t* line ) { fLine = line; }
56  void SetIdx ( Int_t idx ) { fIdx = idx; }
57  void SetNs ( UInt_t ns ) { fNs = ns; }
58  void UpdateErrorCache(Float_t upercache ) { fErrorCache += upercache; }
59 
60  std::vector<Float_t>* GetDataVector() { return &fDataVector; }
61  Float_t GetAlpha() const { return fAlpha; }
62  Float_t GetAlpha_p() const { return fAlpha_p; }
63  Float_t GetDeltaAlpha() const { return fAlpha - fAlpha_p; }
64 
65  Float_t GetErrorCache() const { return fErrorCache; }
66  Int_t GetTypeFlag() const { return fTypeFlag; }
67  Int_t GetNVar() const { return fNVar; }
68  Int_t GetIdx() const { return fIdx;}
69  Float_t* GetLine() const { return fLine;}
70  UInt_t GetNs() const { return fNs;}
71  Float_t GetCweight() const { return fCweight;}
72  Float_t GetTarget() const { return fTarget;}
73 
74  Bool_t IsInI0a() const { return (0.< fAlpha) && (fAlpha<fCweight); }
75  Bool_t IsInI0b() const { return (0.< fAlpha) && (fAlpha_p<fCweight); }
76  Bool_t IsInI0() const { return (IsInI0a() || IsInI0b()); }
77  Bool_t IsInI1() const { return (fAlpha == 0. && fAlpha_p == 0.); }
78  Bool_t IsInI2() const { return (fAlpha == 0. && fAlpha_p == fCweight); }
79  Bool_t IsInI3() const { return (fAlpha == fCweight && fAlpha_p == 0.); }
80 
81  void Print( std::ostream& os ) const;
82  void PrintData();
83 
84  private:
85 
86  std::vector<Float_t> fDataVector;
87  const Float_t fCweight; // svm cost weight
88  Float_t fAlpha; // Lagrange multiplier
89  Float_t fAlpha_p; // Lagrange multiplier
90  Float_t fErrorCache; // optimization parameter
91  UInt_t fNVar; // number of variables
92  const Int_t fTypeFlag; // is sig or bkg - svm requieres 1 for sig and -1 for bkg
93  Int_t fIdx; // index flag
94  UInt_t fNs; // documentation
95  Int_t fIsShrinked; // shrinking flag, see documentation
96  Float_t* fLine; // pointer to column of kernel matrix
97  const Float_t fTarget; // regression target
98 
99  ClassDef(SVEvent,0); // Event for SVM
100  };
101 }
102 
103 #endif //ROOT_TMVA_SVEvent