logistic_regression.h

Go to the documentation of this file.

#ifndef META_LOGISTIC_REGRESSION_H_
#define META_LOGISTIC_REGRESSION_H_

#include "classify/classifier_factory.h"
#include "classify/classifier/classifier.h"
#include "classify/classifier/sgd.h"
#include "index/forward_index.h"
#include "meta.h"

namespace meta
{
namespace classify
{

class logistic_regression : public classifier
{
  public:
    logistic_regression(const std::string& prefix,
                        std::shared_ptr<index::forward_index> idx,
                        double alpha = sgd::default_alpha,
                        double gamma = sgd::default_gamma,
                        double bias = sgd::default_bias,
                        double lambda = sgd::default_lambda,
                        uint64_t max_iter = sgd::default_max_iter);

    std::unordered_map<class_label, double> predict(doc_id d_id);

    virtual class_label classify(doc_id d_id) override;

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

    virtual void reset() override;

    const static std::string id;

  private:
    std::unordered_map<class_label, sgd> classifiers_;
    class_label pivot_;
};

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