src/IO/InputData.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 __INPUT_DATA_H
#define __INPUT_DATA_H

#include <vector>
#include <map> // for class mappings
#include <utility> // for pair
#include <fstream> // for I/O

#include "IO/ClassMappings.h"
#include "Utils/Args.h"

using namespace std;

namespace MultiBoost {


// A couple of useful typedefs
typedef vector< pair<int, double> >::iterator       vpIterator; 
typedef vector< pair<int, double> >::const_iterator cvpIterator; 

enum eInputType
{
   IT_TRAIN, 
   IT_TEST 
};

class InputData
{
public:

   InputData() : _numColumns(0), _numExamples(0),
                 _hasFileName(false), _classInLastColumn(false) {}

   virtual ~InputData();

   virtual void initOptions(nor_utils::Args& args);

   virtual void load(const string& fileName, const eInputType inputType = IT_TRAIN, 
                     const int verboseLevel = 1);

   virtual const int  getClass(const int idx) const { return _data[idx].classIdx; }

   virtual const double getValue(const int idx, const int columnIdx) const {return _data[idx].pValues[columnIdx]; }

   virtual const int  getBinaryClass(const int idx, const int classIdx) const 
                                    { return _data[idx].classIdx == classIdx ? 1: -1;}

   virtual double   getWeight(const int idx, const int classIdx) const { return _data[idx].weights[classIdx]; }

   virtual void setWeight(const int idx, const int classIdx, const double value) 
                          { _data[idx].weights[classIdx] = value; }

   int      getNumColumns() const { return _numColumns; }   
   int      getNumExamples() const { return _numExamples; } 

   int      getNumExamplesPerClass(const int classIdx) const { return _nExamplesPerClass[classIdx]; }

protected:

   int      _numColumns;   
   int      _numExamples;  
   vector<int>   _nExamplesPerClass;   

   bool     _hasFileName;         
   bool     _classInLastColumn;   

   virtual void  initWeights();

#if MB_DEBUG
   void checkVariances(); 
#endif

   // --------------------------------------------------------------------

   struct Example
   {
      Example(double* pValues, const int classIdx, const string& fileName = "") : 
              pValues(pValues), classIdx(classIdx), fileName(fileName) {}

      double*        pValues; 
      vector<double> weights; 
      int            classIdx; 
      string         fileName; 
   };


   vector<Example>   _data;       

};

} // end of namespace MultiBoost

#endif // __INPUT_DATA_H

---