root/TMVA/PDEFoamDecisionTree.h

0001 // @(#)root/tmva $Id$
0002 // Author: Alexander Voigt
0003
0004 /**********************************************************************************
0005  * Project: TMVA - a Root-integrated toolkit for multivariate data analysis       *
0006  * Package: TMVA                                                                  *
0007  * Classes: PDEFoamDecisionTree                                                   *
0008  *                                             *
0009  *                                                                                *
0010  * Description:                                                                   *
0011  *      Class for decision tree like PDEFoam.  It overrides                       *
0012  *      PDEFoam::Explore() to use the decision tree like cell split               *
0013  *      algorithm, given a specific separation type.                              *
0014  *                                                                                *
0015  * Authors (alphabetical):                                                        *
0016  *      S. Jadach        - Institute of Nuclear Physics, Cracow, Poland           *
0017  *      Tancredi Carli   - CERN, Switzerland                                      *
0018  *      Dominik Dannheim - CERN, Switzerland                                      *
0019  *      Alexander Voigt  - TU Dresden, Germany                                    *
0020  *                                                                                *
0021  * Copyright (c) 2010:                                                            *
0022  *      CERN, Switzerland                                                         *
0023  *      MPI-K Heidelberg, Germany                                                 *
0024  *                                                                                *
0025  * Redistribution and use in source and binary forms, with or without             *
0026  * modification, are permitted according to the terms listed in LICENSE           *
0027  * (see tmva/doc/LICENSE)                                          *
0028  **********************************************************************************/
0029
0030 #ifndef ROOT_TMVA_PDEFoamDecisionTree
0031 #define ROOT_TMVA_PDEFoamDecisionTree
0032
0033 #include "TMVA/PDEFoamDiscriminant.h"
0034 #include "TMVA/SeparationBase.h"
0035
0036 namespace TMVA
0037 {
0038
0039    class PDEFoamDecisionTree : public PDEFoamDiscriminant
0040    {
0041
0042    private:
0043       SeparationBase *fSepType;    // separation type
0044
0045    protected:
0046
0047       virtual void Explore(PDEFoamCell *Cell);     // Exploration of the cell
0048
0049       PDEFoamDecisionTree(const PDEFoamDecisionTree&); // Copy Constructor  NOT USED
0050
0051    public:
0052       PDEFoamDecisionTree();               // Default constructor (used only by ROOT streamer)
0053       PDEFoamDecisionTree(const TString&, SeparationBase *sepType, UInt_t cls); // Principal user-defined constructor
0054       virtual ~PDEFoamDecisionTree();      // Default destructor
0055
0056       // ---------- ROOT class definition
0057       ClassDef(PDEFoamDecisionTree, 1) // Decision tree like PDEFoam
0058          }; // end of PDEFoamDecisionTree
0059
0060 }  // namespace TMVA
0061
0062 #endif