model.h

Go to the documentation of this file.

#ifndef __ZGEN_SHIFTREDUCE_MODEL_MODEL_H__
#define __ZGEN_SHIFTREDUCE_MODEL_MODEL_H__

#include "shiftreduce/option.h"
#include "score.h"
#include "score_map.h"
#include <vector>

namespace ZGen {
namespace ShiftReduce {

class Model {
public:
  Model(const option_t& opts);

  floatval_t score(const StateItem& state, const Action& act, bool avg);

  floatval_t score(const ScoreContext& ctx, const Action& act, bool avg);

  void update(const StateItem& state, const Action& act, int timestamp,
      floatval_t scale);

  void update(const ScoreContext& ctx, const Action& act, int timestamp,
      floatval_t scale);

  void flush(int timestamp);

  bool save(std::ostream& os);

  bool load(std::istream& is);
private:
  std::vector< ScoreMap<UnigramScore> > uscore_repo;
  std::vector< ScoreMap<BigramScore>  > bscore_repo;
  std::vector< ScoreMap<TrigramScore> > tscore_repo;

  std::vector< UnigramScore > uscore_cache;
  std::vector< BigramScore >  bscore_cache;
  std::vector< TrigramScore > tscore_cache;
};

}
}

#endif  //  end for __ZGEN_SHIFTREDUCE_MODEL_MODEL_H__