PhysicsTools::TreeTrainer Class Reference

#include <TreeTrainer.h>

Public Member Functions
void	addReader (const TreeReader &reader)
void	addTree (TTree *tree, int target=-1, double weight=-1.0)
bool	iteration (MVATrainer *trainer)
void	reset ()
Calibration::MVAComputer *	train (const std::string &trainDescription, double crossValidation=0.0, bool useXSLT=false)
void	train (MVATrainer *trainer)
	TreeTrainer ()
	TreeTrainer (TTree *tree, double weight=-1.0)
	TreeTrainer (TTree *signal, TTree *background, double weight=-1.0)
	~TreeTrainer ()

Private Attributes
std::vector< TreeReader >	readers
std::vector< double * >	weights

Detailed Description

Definition at line 18 of file TreeTrainer.h.

Constructor & Destructor Documentation

PhysicsTools::TreeTrainer::TreeTrainer

(

)

Definition at line 19 of file TreeTrainer.cc.

{
}

PhysicsTools::TreeTrainer::TreeTrainer	(	TTree *	tree,
		double	weight = -1.0
	)

Definition at line 23 of file TreeTrainer.cc.

References addTree().

{
    addTree(tree, -1, weight);
}

PhysicsTools::TreeTrainer::TreeTrainer	(	TTree *	signal,
		TTree *	background,
		double	weight = -1.0
	)

Definition at line 28 of file TreeTrainer.cc.

References addTree().

{
    addTree(signal, true, weight);
    addTree(background, false, weight);
}

PhysicsTools::TreeTrainer::~TreeTrainer

(

)

Definition at line 34 of file TreeTrainer.cc.

References reset().

{
    reset();
}

Member Function Documentation

void PhysicsTools::TreeTrainer::addReader

(

const TreeReader &

reader

)

Definition at line 82 of file TreeTrainer.cc.

References readers.

Referenced by addTree().

{
    readers.push_back(reader);
}

void PhysicsTools::TreeTrainer::addTree	(	TTree *	tree,
		int	target = -1,
		double	weight = -1.0
	)

Definition at line 58 of file TreeTrainer.cc.

References addReader(), PhysicsTools::TreeReader::addSingle(), PhysicsTools::MVATrainer::kTargetId, PhysicsTools::MVATrainer::kWeightId, matplotRender::reader, and weights.

Referenced by TreeTrainer().

{
    static const bool targets[2] = { true, false };

    TreeReader reader(tree, false, weight > 0.0);

    if (target >= 0) {
        if (tree->GetBranch("__TARGET__"))
            throw cms::Exception("TreeTrainer")
                << "__TARGET__ branch already present in file."
                << std::endl;

        reader.addSingle(MVATrainer::kTargetId, &targets[!target]);
    }

    if (weight > 0.0) {
        double *ptr = new double(weight);
        weights.push_back(ptr);
        reader.addSingle(MVATrainer::kWeightId, ptr);
    }

    addReader(reader);
}

bool PhysicsTools::TreeTrainer::iteration

(

MVATrainer *

trainer

)

Definition at line 87 of file TreeTrainer.cc.

References calib, PhysicsTools::MVATrainer::getTrainCalibration(), PhysicsTools::TreeReader::loop(), and readers.

Referenced by train().

{
    Calibration::MVAComputer *calib = trainer->getTrainCalibration();   
    if (!calib)
        return true;

    MVAComputer computer(calib, true);

    std::for_each(readers.begin(), readers.end(),
                  std::bind2nd(std::mem_fun_ref(&TreeReader::loop),
                               &computer));

    return false;
}

void PhysicsTools::TreeTrainer::reset

(

void

)

Definition at line 50 of file TreeTrainer.cc.

References readers, and weights.

Referenced by ~TreeTrainer().

{
    readers.clear();
    std::for_each(weights.begin(), weights.end(),
                  std::ptr_fun(&::operator delete));
    weights.clear();
}

Calibration::MVAComputer * PhysicsTools::TreeTrainer::train	(	const std::string &	trainDescription,
		double	crossValidation = 0.0,
		bool	useXSLT = false
	)

Definition at line 39 of file TreeTrainer.cc.

References PhysicsTools::MVATrainer::getCalibration(), PhysicsTools::MVATrainer::setCrossValidation(), and PhysicsTools::MVATrainer::setMonitoring().

{
    MVATrainer trainer(trainFile, useXSLT);
    trainer.setMonitoring(true);
    trainer.setCrossValidation(crossValidation);
    train(&trainer);
    return trainer.getCalibration();
}

void PhysicsTools::TreeTrainer::train

(

MVATrainer *

trainer

)

Definition at line 102 of file TreeTrainer.cc.

References iteration().

{
    while(!iteration(trainer));
}

Member Data Documentation

std::vector<TreeReader> PhysicsTools::TreeTrainer::readers

private

Definition at line 40 of file TreeTrainer.h.

Referenced by addReader(), iteration(), and reset().

std::vector<double*> PhysicsTools::TreeTrainer::weights

private

Definition at line 42 of file TreeTrainer.h.

Referenced by addTree(), and reset().