src/IO/ClassMappings.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
*
* 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 __CLASS_MAPPINGS_H
#define __CLASS_MAPPINGS_H

#include <string>
#include <vector>
#include <map>

using namespace std;


namespace MultiBoost {

class ClassMappings
{
public:

   static int addClassName(const string& className);

   static void loadClassMapFile(const string& classMapFileName);

   static string getClassNameFromIdx(const int idx); 

   static int    getIdxFromClassName(const string& className);

   static int    getNumClasses() { return _numRegClasses; }   

private:
   static vector<string>   _mapIdxToClass; 

   static map<string, int>  _mapClassToIdx;

   static int _numRegClasses; 
};

} // end of namespace MultiBoost

#endif // CLASS_MAPPINGS_H

---