1 |
dkralph |
1.1 |
#include <iostream>
|
2 |
|
|
#include <sstream>
|
3 |
|
|
|
4 |
|
|
#include "TString.h"
|
5 |
|
|
|
6 |
|
|
#include "TMVA/Reader.h"
|
7 |
|
|
#include "TMVA/Tools.h"
|
8 |
|
|
#include "TMVA/Config.h"
|
9 |
|
|
#include "TMVA/MethodBDT.h"
|
10 |
|
|
|
11 |
|
|
#include "filestuff.h"
|
12 |
|
|
|
13 |
|
|
using namespace std;
|
14 |
|
|
|
15 |
|
|
class BdtInfo
|
16 |
|
|
{
|
17 |
|
|
public:
|
18 |
|
|
BdtInfo(TString varStr, TString weightFile, filestuff *fs, bool translatePar=false);
|
19 |
|
|
void setVar(TString name, filestuff *fs);
|
20 |
|
|
void setSpec(TString name, filestuff *fs);
|
21 |
|
|
void setVals(filestuff *fs);
|
22 |
|
|
TString trans(TString name); // names changed at some point
|
23 |
|
|
|
24 |
|
|
TMVA::Reader reader;
|
25 |
|
|
vector<TString> vars,specs;
|
26 |
|
|
map<TString,float*> vals;
|
27 |
|
|
bool translate;
|
28 |
|
|
};
|