one_vs_all.h

Go to the documentation of this file.

#ifndef META_CLASSIFY_ONE_VS_ALL_H_
#define META_CLASSIFY_ONE_VS_ALL_H_

#include "classify/classifier/binary_classifier.h"
#include "classify/classifier_factory.h"
#include "meta.h"

namespace meta
{
namespace classify
{

class one_vs_all : public classifier
{
  public:
    template <class Function>
    one_vs_all(std::shared_ptr<index::forward_index> idx, Function&& create)
        : classifier{std::move(idx)}
    {
        for (const auto& label : idx_->class_labels())
            classifiers_.emplace(label, create(label));
    }

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

    class_label classify(doc_id d_id) override;

    void reset() override;

    const static std::string id;

  private:
    std::unordered_map<class_label, std::unique_ptr<binary_classifier>>
        classifiers_;
};

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