blob: 76b4f6ef731162a9ff85d6b25f468f78eb0d10b2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
#ifndef moses_TargetBigramFeature_h
#define moses_TargetBigramFeature_h
#include <string>
#include <map>
#include "FactorCollection.h"
#include "FeatureFunction.h"
#include "FFState.h"
#include "Word.h"
namespace Moses
{
class TargetBigramState : public FFState {
public:
TargetBigramState(const Word& word): m_word(word) {}
const Word& GetWord() const {return m_word;}
virtual int Compare(const FFState& other) const;
private:
Word m_word;
};
/** Sets the features of observed bigrams.
*/
class TargetBigramFeature : public StatefulFeatureFunction {
public:
TargetBigramFeature(FactorType factorType = 0):
StatefulFeatureFunction("dlmb", ScoreProducer::unlimited),
m_factorType(factorType)
{
FactorCollection& factorCollection = FactorCollection::Instance();
const Factor* bosFactor =
factorCollection.AddFactor(Output,m_factorType,BOS_);
m_bos.SetFactor(m_factorType,bosFactor);
}
bool Load(const std::string &filePath);
std::string GetScoreProducerWeightShortName(unsigned) const;
size_t GetNumInputScores() const;
virtual const FFState* EmptyHypothesisState(const InputType &input) const;
virtual FFState* Evaluate(const Hypothesis& cur_hypo, const FFState* prev_state,
ScoreComponentCollection* accumulator) const;
virtual FFState* EvaluateChart( const ChartHypothesis& /* cur_hypo */,
int /* featureID */,
ScoreComponentCollection* ) const
{
abort();
}
private:
FactorType m_factorType;
Word m_bos;
std::set<std::string> m_vocab;
};
}
#endif // moses_TargetBigramFeature_h
|