Trainer.hh

Go to the documentation of this file.

#pragma once

#include "TFile.h"

#include "TMVA/Factory.h"
// #include "TMVA/DataLoader.h"

#include "boca/identification/Tag.hh"

namespace boca {

namespace tagger{
class Base;
}

class Trainer {

public:

    Trainer(tagger::Base& tagger);

private:

    std::vector<double> BookMethod();

    void AddObservables();

    long AddAllTrees();

    long AddTrees(Tag tag);

    void PrepareTrainingAndTestTree(long event_number);

    long AddTree(std::string const& tree_name, Tag tag);

    long Entries(std::string const& tree_name, Tag tag);

    double Weight(std::string const& tree_name, Tag tag);

    std::string FactoryOptions();

    std::string DataLoaderOptions();

    std::string MethodOptions() const;

    tagger::Base const& Tagger() const;

    tagger::Base & Tagger();

    TMVA::Factory const& Factory() const;

    TMVA::Factory& Factory();

    TMVA::Factory & DataLoader();

//     TMVA::DataLoader data_loader_;

    tagger::Base& tagger_;

    TFile output_;

    TMVA::Factory factory_;

    std::map<Tag, TFile> input_;

};

}