diff options
Diffstat (limited to 'decoder/ff_wordalign.cc')
| -rw-r--r-- | decoder/ff_wordalign.cc | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/decoder/ff_wordalign.cc b/decoder/ff_wordalign.cc index f07eda02..c9f90541 100644 --- a/decoder/ff_wordalign.cc +++ b/decoder/ff_wordalign.cc @@ -144,6 +144,15 @@ SourcePOSBigram::SourcePOSBigram(const std::string& param) : cerr << " (" << pos_.size() << " lines)\n"; } +void SourcePOSBigram::FinalTraversalFeatures(const void* context, + SparseVector<double>* features) const { + WordID left = *static_cast<const WordID*>(context); + int left_wc = *(static_cast<const int*>(context) + 1); + if (left_wc == 1) + FireFeature(-1, left, features); + FireFeature(left, -1, features); +} + void SourcePOSBigram::FireFeature(WordID left, WordID right, SparseVector<double>* features) const { @@ -187,8 +196,6 @@ void SourcePOSBigram::TraversalFeaturesImpl(const SentenceMetadata& smeta, FireFeature(-1, left, features); FireFeature(left, right, features); out_word_count = left_wc + right_wc; - if (out_word_count == smeta.GetSourceLength()) - FireFeature(right, -1, features); out_context = right; } } |