svm_wrapper.h

Go to the documentation of this file.

#ifndef META_SVM_WRAPPER_H_
#define META_SVM_WRAPPER_H_

#include <unordered_map>
#include "classify/classifier_factory.h"
#include "classify/classifier/classifier.h"
#include "index/forward_index.h"
#include "meta.h"

namespace meta
{
namespace classify
{

class svm_wrapper : public classifier
{
  public:
    enum kernel
    {
        None,
        Quadratic,
        Cubic,
        Quartic,
        RBF,
        Sigmoid
    };

    svm_wrapper(std::shared_ptr<index::forward_index> idx,
                const std::string& svm_path, kernel kernel_opt = kernel::None);

    class_label classify(doc_id d_id) override;

    void train(const std::vector<doc_id>& docs) override;

    confusion_matrix test(const std::vector<doc_id>& docs) override;

    void reset() override;

    const static std::string id;

  private:
    const std::string svm_path_;

    const static std::unordered_map<kernel, std::string, std::hash<int>>
        options_;

    kernel kernel_;

    std::string executable_;
};

template <>
std::unique_ptr<classifier>
    make_classifier<svm_wrapper>(const cpptoml::table&,
                                 std::shared_ptr<index::forward_index>);
}
}

#endif