model.h

Go to the documentation of this file.

#ifndef __ZGEN_BESTFRIST_MODEL_H__
#define __ZGEN_BESTFRIST_MODEL_H__

#include "settings.h"
#include "bestfirst/search/span.h"
#include "bestfirst/model/score_map.h"

namespace ZGen {

namespace BestFirst {

class Model {
public:
  Model();

  floatval_t score(const Span& span, bool avg) const;

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

  void update(const Span& span, int timestamp, floatval_t scale);

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

  void flush(int timestamp);

  bool load(std::istream& is);

  bool save(std::ostream& os);
private:
  std::vector< ScoreMap<UnigramScore> > uscore_repo;
  std::vector< ScoreMap<BigramScore> >  bscore_repo;
  std::vector< ScoreMap<TrigramScore> > tscore_repo;
};

} //  end for namespace Bestfirst
} //  end for namespace ZGen


#endif  //  end for __ZGEN_BESTFRIST_MODEL_H__