src/AdaBoostMHLearner.h

Go to the documentation of this file.

/*
* This file is part of MultiBoost, a multi-class 
* AdaBoost learner/classifier
*
* Copyright (C) 2005 Norman Casagrande
* For informations write to nova77@gmail.com
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*
*/

#ifndef __ADA_BOOST_H
#define __ADA_BOOST_H

#include "IO/InputData.h"
#include "Utils/Args.h"
#include "WeakLearners/BaseLearner.h"

using namespace std;


namespace MultiBoost {

class AdaBoostLearner
{
public:

   AdaBoostLearner(nor_utils::Args& args, int verbose = 1);

   void run(int numIterations, const string& trainFileName, const string& testFileName = "");
   
protected:

   double updateWeights(InputData* pTrainData, BaseLearner* pWeakHypothesis);

   vector<BaseLearner*>  _foundHypotheses; 

   InputData*  _pTrainData; 
   InputData*  _pTestData; 

   string      _basicLearnerName; 
   string      _shypFileName; 

   int         _maxTime; 
   double      _theta; 

   nor_utils::Args&  _args;  

   int         _verbose;
   string      _outputInfoFile; 

   const double _smallVal; 

   
};

} // end of namespace MultiBoost

#endif // __ADA_BOOST_H

---