binary_classifier.h

Go to the documentation of this file.

#ifndef META_BINARY_CLASSIFIER_H_
#define META_BINARY_CLASSIFIER_H_

#include "classify/classifier/classifier.h"

namespace meta
{
namespace classify
{

class binary_classifier : public classifier
{
  public:
    binary_classifier(std::shared_ptr<index::forward_index> idx,
                      class_label positive, class_label negative);

    class_label classify(doc_id d_id) final;

    virtual double predict(doc_id d_id) const = 0;

    const class_label& positive_label() const;

    const class_label& negative_label() const;

  private:
    const class_label positive_;

    const class_label negative_;
};

}
}
#endif