svm_classifier.hpp

Go to the documentation of this file.

// Copyright (c) 2012, 2019 Scott Niekum, Joshua Whitley
// All rights reserved.
//
// Software License Agreement (BSD License 2.0)
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions
// are met:
//
//  * Redistributions of source code must retain the above copyright
//    notice, this list of conditions and the following disclaimer.
//  * Redistributions in binary form must reproduce the above
//    copyright notice, this list of conditions and the following
//    disclaimer in the documentation and/or other materials provided
//    with the distribution.
//  * Neither the name of {copyright_holder} nor the names of its
//    contributors may be used to endorse or promote products derived
//    from this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
// FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
// COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
// INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
// BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
// LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
// CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
// LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
// ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.

#ifndef ML_CLASSIFIERS__SVM_CLASSIFIER_HPP_
#define ML_CLASSIFIERS__SVM_CLASSIFIER_HPP_

#include "libsvm-3.14/svm.h"

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

#include "ml_classifiers/classifier.hpp"

namespace ml_classifiers
{

using CPoint = std::vector<double>;
using CPointList = std::vector<CPoint>;
using ClassMap = std::map<std::string, CPointList>;

class SVMClassifier : public Classifier
{
public:
  ClassMap class_data;
  svm_problem svm_data;
  svm_model * trained_model;
  std::map<std::string, int> label_str_to_int;
  std::map<int, std::string> label_int_to_str;
  double ** scaling_factors;

  SVMClassifier();
  ~SVMClassifier();

  void save(const std::string filename);
  bool load(const std::string filename);
  void addTrainingPoint(std::string target_class, const std::vector<double> point);
  void train();
  void clear();
  std::string classifyPoint(const std::vector<double> point);
};

}  // namespace ml_classifiers

#endif  // ML_CLASSIFIERS__SVM_CLASSIFIER_HPP_