From a046645ca3e2ac1ac8839ba2856c49bd771be62f Mon Sep 17 00:00:00 2001
From: Patrick Simianer <p@simianer.de>
Date: Thu, 5 Nov 2015 20:17:40 +0100
Subject: dtrain_net_interface output rules too

---
 training/dtrain/sample_net_interface.h | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'training/dtrain/sample_net_interface.h')

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
-- 
cgit v1.2.3