binary_classifier_factory.h

Go to the documentation of this file.

#ifndef META_BINARY_CLASSIFIER_FACTORY_H_
#define META_BINARY_CLASSIFIER_FACTORY_H_

#include "classify/classifier/binary_classifier.h"
#include "util/factory.h"

namespace cpptoml
{
class table;
}

namespace meta
{
namespace classify
{

class binary_classifier_factory
    : public util::factory<binary_classifier_factory, binary_classifier,
                           const cpptoml::table&,
                           std::shared_ptr<index::forward_index>, class_label,
                           class_label>
{
    friend base_factory;

  private:
    binary_classifier_factory();

    template <class Classifier>
    void reg();
};

// doxygen is being stupid here, so I'm forced to specify which overload
// I'm talking about in their brief...
std::unique_ptr<binary_classifier>
    make_binary_classifier(const cpptoml::table& config,
                           std::shared_ptr<index::forward_index> idx,
                           class_label positive, class_label negative);

template <class Classifier>
std::unique_ptr<binary_classifier>
    make_binary_classifier(const cpptoml::table&,
                           std::shared_ptr<index::forward_index> idx,
                           class_label positive, class_label negative)
{
    return make_unique<Classifier>(std::move(idx), positive, negative);
}

template <class Classifier>
void register_binary_classifier()
{
    binary_classifier_factory::get().add(Classifier::id,
                                         make_binary_classifier<Classifier>);
}
}
}
#endif