perceptron.h

#ifndef META_SEQUENCE_PERCEPTRON_H_
#define META_SEQUENCE_PERCEPTRON_H_

#include <random>

#include "classify/models/linear_model.h"
#include "sequence/sequence_analyzer.h"

namespace meta
{
namespace sequence
{

class perceptron
{
  public:
    struct training_options
    {
        uint64_t max_iterations = 5;

        std::random_device::result_type seed = std::random_device{}();
    };

    perceptron();

    perceptron(const std::string& prefix);

    void tag(sequence& seq) const;

    void train(std::vector<sequence>& sequences,
               const training_options& options);

    void save(const std::string& prefix) const;

  private:
    sequence_analyzer analyzer_;

    classify::linear_model<feature_id, double, label_id> model_;
};
}
}

#endif