Welcome to mirror list, hosted at ThFree Co, Russian Federation.

SparseHieroReorderingFeature.h « FF « moses - github.com/moses-smt/mosesdecoder.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 79fb0f19b7e5bcb5ffe633a57c042693395936ba (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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
#pragma once

#include <string>

#include <boost/unordered_set.hpp>

#include <util/string_piece.hh>

#include "moses/Factor.h"
#include "moses/Sentence.h"

#include "StatelessFeatureFunction.h"

namespace Moses
{

class SparseHieroReorderingFeature : public StatelessFeatureFunction
{
public:
  enum Type {
    SourceCombined,
    SourceLeft,
    SourceRight
  };

  SparseHieroReorderingFeature(const std::string &line);

  bool IsUseable(const FactorMask &mask) const {
    return true;
  }

  void SetParameter(const std::string& key, const std::string& value);

  void EvaluateInIsolation(const Phrase &source
                           , const TargetPhrase &targetPhrase
                           , ScoreComponentCollection &scoreBreakdown
                           , ScoreComponentCollection &estimatedScores) const {
  }
  virtual void EvaluateWithSourceContext(const InputType &input
                                         , const InputPath &inputPath
                                         , const TargetPhrase &targetPhrase
                                         , const StackVec *stackVec
                                         , ScoreComponentCollection &scoreBreakdown
                                         , ScoreComponentCollection *estimatedScores = NULL)  const {
  }

  void EvaluateTranslationOptionListWithSourceContext(const InputType &input
      , const TranslationOptionList &translationOptionList) const {
  }

  virtual void EvaluateWhenApplied(const Hypothesis& hypo,
                                   ScoreComponentCollection* accumulator) const {
  }
  void EvaluateWhenApplied(const ChartHypothesis &hypo,
                           ScoreComponentCollection* accumulator) const;


private:

  typedef boost::unordered_set<const Factor*> Vocab;

  void AddNonTerminalPairFeatures(
    const Sentence& sentence, const Range& nt1, const Range& nt2,
    bool isMonotone, ScoreComponentCollection* accumulator) const;

  void LoadVocabulary(const std::string& filename, Vocab& vocab);
  const Factor*  GetFactor(const Word& word, const Vocab& vocab, FactorType factor) const;

  Type m_type;
  FactorType m_sourceFactor;
  FactorType m_targetFactor;
  std::string m_sourceVocabFile;
  std::string m_targetVocabFile;

  const Factor* m_otherFactor;

  Vocab m_sourceVocab;
  Vocab m_targetVocab;

};


}