mosesdecoder/contrib/memscore/phraselm.h
2011-11-23 15:31:16 +00:00

46 lines
957 B
C++

// memscore - in-memory phrase scoring for Statistical Machine Translation
// Christian Hardmeier, FBK-irst, Trento, 2010
// $Id$
#ifndef PHRASELM_H
#define PHRASELM_H
#include <cassert>
#include "memscore.h"
#include "phrasetable.h"
#include "statistic.h"
class lmtable;
class PhraseLanguageModel : public PhraseStatistic
{
protected:
String lmfile_;
Count score_idx_;
PhraseInfoList *phrase_info_list_;
void compute_lmscores(PhraseInfoList &phrase_info_list, bool closed_world);
public:
PhraseLanguageModel(String lmfile) : lmfile_(lmfile) {}
virtual void attach(PhraseInfoList &pilist);
virtual void compute_statistic();
virtual Score get_score(PhraseInfo &pi) {
assert(computation_done_);
return pi.data(score_idx_);
}
};
class ClosedPhraseLanguageModel : public PhraseLanguageModel
{
public:
ClosedPhraseLanguageModel(String lmfile) : PhraseLanguageModel(lmfile) {}
virtual void compute_statistic();
};
#endif