one_vs_one.h

Go to the documentation of this file.

#ifndef META_CLASSIFY_ONE_VS_ONE_H_
#define META_CLASSIFY_ONE_VS_ONE_H_

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

namespace meta
{
namespace classify
{

class one_vs_one : public classifier
{
  public:
    template <class Function>
    one_vs_one(std::shared_ptr<index::forward_index> idx, Function&& create)
        : classifier{std::move(idx)}
    {
        auto labels = idx_->class_labels();
        for (uint64_t i = 0; i < labels.size(); ++i)
        {
            for (uint64_t j = i + 1; j < labels.size(); ++j)
            {
                classifiers_.emplace_back(create(labels[i], labels[j]));
            }
        }
    }

    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::vector<std::unique_ptr<binary_classifier>> classifiers_;
};

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