diff options
| author | Patrick Simianer <p@simianer.de> | 2015-11-05 20:17:40 +0100 |
|---|---|---|
| committer | Patrick Simianer <p@simianer.de> | 2015-11-05 20:17:40 +0100 |
| commit | a046645ca3e2ac1ac8839ba2856c49bd771be62f (patch) | |
| tree | 192fb60187b9d4b6c761136ebd8ae595563ab6d7 /training/dtrain/sample_net_interface.h | |
| parent | e5504d3f86c6a9a0763540d95503246e033ade13 (diff) | |
dtrain_net_interface output rules too
Diffstat (limited to 'training/dtrain/sample_net_interface.h')
| -rw-r--r-- | training/dtrain/sample_net_interface.h | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/training/dtrain/sample_net_interface.h b/training/dtrain/sample_net_interface.h index a2b5f87d..6d00e5d5 100644 --- a/training/dtrain/sample_net_interface.h +++ b/training/dtrain/sample_net_interface.h @@ -17,7 +17,7 @@ struct ScoredKbest : public DecoderObserver vector<Ngrams>* ref_ngs_; vector<size_t>* ref_ls_; bool dont_score; - string viterbiTreeStr_; + string viterbiTreeStr_, viterbiRules_; ScoredKbest(const size_t k, PerSentenceBleuScorer* scorer) : k_(k), scorer_(scorer), dont_score(false) {} @@ -44,6 +44,9 @@ struct ScoredKbest : public DecoderObserver effective_sz_++; feature_count_ += h.f.size(); viterbiTreeStr_ = hg->show_viterbi_tree(false); + ostringstream ss; + ViterbiRules(*hg, &ss); + viterbiRules_ = ss.str(); } } @@ -56,6 +59,7 @@ struct ScoredKbest : public DecoderObserver inline size_t GetFeatureCount() { return feature_count_; } inline size_t GetSize() { return effective_sz_; } inline string GetViterbiTreeStr() { return viterbiTreeStr_; } + inline string GetViterbiRules() { return viterbiRules_; } }; } // namespace |