MAClassifierModels.hpp

/*
 *  This file is part of the AiBO+ project
 *
 *  Copyright (C) 2005-2016 Csaba Kertész (csaba.kertesz@gmail.com)
 *
 *  AiBO+ is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  AiBO+ 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 General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Street #330, Boston, MA 02111-1307, USA.
 *
 */

#pragma once

#include <string>

#include <boost/unordered_map.hpp>

#include "ml/MAClassifier.hpp"
#include "ml/MAModel.hpp"

namespace MA
{
#if !defined(__AIBO_BUILD__)
typedef boost::unordered_map<std::string, ClassifierSPtr> StringClassifierSPtrMap;
#endif

typedef boost::unordered_map<std::string, ModelSPtr> StringModelSPtrMap;

static const std::string FloorSurfacesID = "FloorSurfaces";
static const std::string FloorSurfaceVectorID = "FloorSurfaceVector";
static const std::string RecognizedFloorSurfaceID = "RecognizedFloorSurface";
static const std::string RealFloorSurfaceID = "RealSurface";
static const std::string UserDefinedFloorSurfaceID = "UserDefinedFloorSurface";
static const std::string FloorSurfaceRemarkID = "FloorSurfaceRemark";
static const std::string DataCollectorDogSocksID = "DataCollectorDogSocks";
static const std::string DataCollectorLocationID = "DataCollectorLocation";
static const std::string DataCollectorEmailID = "DataCollectorEmail";
static const std::string DataCollectorUsernameID = "DataCollectorUsername";
static const std::string DataCollectorEaID = "DataCollectorEmotionalAttachment";
static const std::string FloorSurfaceModelID = "FloorSurfaceModel";
static const std::string BodyStateID = "BodyState";
static const std::string BodyStateModelID = "BodyStateModel";
static const std::string SoundEventModelID = "SoundEventModel";
// The maximum wait time is 3 minutes for user feedback (in msec)
static const int MaxFeedbackWaitTime = 3*60*1000;
}

/*
 * @brief Get the feature vector partitions to create a weak model
 *
 * @param model_name Model name
 *
 * @return Feature vector partition sizes
 *
 */

extern MC::IntList MAGetWeakModelPartitions(const std::string& model_name);

class MAClassifierModels
{
public:

  MAClassifierModels();

private:

  /*
   * @brief Load a model from a file
   *
   * @param file_name File name
   * @param model_name Model name
   *
   */

  void LoadModelFromFile(const std::string& file_name, const std::string& model_name);

public:

  /*
   * @brief Recognize a feature vector with a model
   *
   * @param model_name Model name
   * @param feature_vector Feature vector to be recognized
   * @param[out] confidence Confidence value for the prediction
   *
   * @return Output of the model for the feature vector otherwise infinity
   *
   */

  float Recognize(const std::string& model_name, MC::FloatList& feature_vector,
                  MC::FloatList& confidence);

private:
#if !defined(__AIBO_BUILD__)
  MA::StringClassifierSPtrMap RawModels;
#endif
  MA::StringModelSPtrMap Models;
};