scorer.h

Go to the documentation of this file.

#ifndef META_SEQUENCE_CRF_SCORER_H_
#define META_SEQUENCE_CRF_SCORER_H_

#include "sequence/crf/crf.h"

namespace meta
{
namespace sequence
{

class crf::scorer
{
  public:
    void score(const crf& model, const sequence& seq);

    void transition_scores(const crf& model);

    void state_scores(const crf& model, const sequence& seq);

    void forward();

    void backward();

    void marginals();

    double state(uint64_t time, label_id lbl) const;

    double state_exp(uint64_t time, label_id lbl) const;

    double trans(label_id from, label_id to) const;

    double trans_exp(label_id from, label_id to) const;

    double forward(uint64_t time, label_id lbl) const;

    double backward(uint64_t time, label_id lbl) const;

    double state_marginal(uint64_t time, label_id lbl) const;

    double trans_marginal(label_id from, label_id to) const;

    double loss(const sequence& seq) const;

  private:
    void transition_marginals();

    void state_marginals();

    double_matrix state_;
    double_matrix state_exp_;
    double_matrix trans_;
    double_matrix trans_exp_;

    util::optional<forward_trellis> fwd_;
    util::optional<trellis> bwd_;
    util::optional<double_matrix> state_mrg_;
    util::optional<double_matrix> trans_mrg_;
};

}
}

#endif