diff options
Diffstat (limited to 'decoder/apply_models.cc')
| -rw-r--r-- | decoder/apply_models.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/decoder/apply_models.cc b/decoder/apply_models.cc index 0e83582f..2b518d62 100644 --- a/decoder/apply_models.cc +++ b/decoder/apply_models.cc @@ -96,7 +96,7 @@ struct Candidate { if (is_goal) { assert(tail.size() == 1); const string& ant_state = node_states[tail.front()]; - models.AddFinalFeatures(ant_state, &out_edge_); + models.AddFinalFeatures(ant_state, &out_edge_, smeta); } else { models.AddFeaturesToEdge(smeta, out_hg, node_states, &out_edge_, &state_, &edge_estimate); } @@ -344,7 +344,7 @@ struct NoPruningRescorer { if (is_goal) { assert(tail.size() == 1); const string& ant_state = node_states_[tail.front()]; - models.AddFinalFeatures(ant_state, new_edge); + models.AddFinalFeatures(ant_state, new_edge,smeta); } else { prob_t edge_estimate; // this is a full intersection, so we disregard this models.AddFeaturesToEdge(smeta, out, node_states_, new_edge, &head_state, &edge_estimate); |