root/TMVA/TrainingHistory.h

0001 /**********************************************************************************
0002  * Project: TMVA - a Root-integrated toolkit for multivariate data analysis       *
0003  * Package: TMVA                                                                  *
0004  * Class  : MsgLogger                                                             *
0005  *                                             *
0006  *                                                                                *
0007  * Description:                                                                   *
0008  *      Implementation (see header for description)                               *
0009  *                                                                                *
0010  * Author:                                                                        *
0011  *      Joseph McKenna        <Joseph.McKenna@cern.ch> - Aarhus, Denmark          *
0012  *                                                                                *
0013 * Copyright (c) 2019:                                                             *
0014  *      Aarhus, Denmark                                                           *
0015  *                                                                                *
0016  * Redistribution and use in source and binary forms, with or without             *
0017  * modification, are permitted according to the terms listed in LICENSE           *
0018  * (see tmva/doc/LICENSE)                                          *
0019  **********************************************************************************/
0020
0021
0022 #ifndef ROOT_TMVA_TrainingHistory
0023 #define ROOT_TMVA_TrainingHistory
0024
0025 #include <vector>
0026 #include "TString.h"
0027 #include <map>
0028
0029 namespace TMVA {
0030
0031    class TrainingHistory {
0032
0033    public:
0034       typedef std::vector<std::pair<Int_t,Double_t>> IterationRecord;
0035       TrainingHistory();
0036       virtual ~TrainingHistory();
0037
0038       void AddValue(TString Property, Int_t stage, Double_t value);
0039       void SaveHistory(TString Name);
0040    private:
0041       std::map<TString,int> fHistoryMap;
0042       std::vector<IterationRecord*> fHistoryData;
0043
0044    };
0045
0046 } // namespace TMVA
0047
0048 #endif