From cb9b2004c95a162b9a2e42b3fbe9677fc54693bb Mon Sep 17 00:00:00 2001
From: philblunsom <philblunsom@ec762483-ff6d-05da-a07a-a48fb63a330f>
Date: Fri, 2 Jul 2010 20:36:27 +0000
Subject: Updated pipeline defaults.

git-svn-id: https://ws10smt.googlecode.com/svn/trunk@121 ec762483-ff6d-05da-a07a-a48fb63a330f
---
 gi/pyp-topics/src/train-contexts.cc | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

(limited to 'gi/pyp-topics')

diff --git a/gi/pyp-topics/src/train-contexts.cc b/gi/pyp-topics/src/train-contexts.cc
index de107c86..7e2100f8 100644
--- a/gi/pyp-topics/src/train-contexts.cc
+++ b/gi/pyp-topics/src/train-contexts.cc
@@ -96,7 +96,7 @@ int main(int argc, char **argv)
     ogzstream documents_out(vm["document-topics-out"].as<string>().c_str());
 
     int document_id=0;
-   map<int,int> all_terms;
+    map<int,int> all_terms;
     for (Corpus::const_iterator corpusIt=contexts_corpus.begin(); 
          corpusIt != contexts_corpus.end(); ++corpusIt, ++document_id) {
       vector<int> unique_terms;
@@ -125,13 +125,15 @@ int main(int argc, char **argv)
     }
     documents_out.close();
 
-   ofstream default_topics(vm["default-topics-out"].as<string>().c_str());
-    default_topics << model.max_topic() <<endl;
-    for (std::map<int,int>::const_iterator termIt=all_terms.begin(); termIt != all_terms.end(); ++termIt) {
-     vector<std::string> strings = contexts_corpus.context2string(termIt->first);
-      default_topics << model.max(-1, termIt->first) << " ||| " << termIt->second << " ||| ";
-     copy(strings.begin(), strings.end(),ostream_iterator<std::string>(default_topics, " "));
-      default_topics <<endl;
+    if (vm.count("default-topics-out")) {
+      ofstream default_topics(vm["default-topics-out"].as<string>().c_str());
+      default_topics << model.max_topic() <<endl;
+      for (std::map<int,int>::const_iterator termIt=all_terms.begin(); termIt != all_terms.end(); ++termIt) {
+       vector<std::string> strings = contexts_corpus.context2string(termIt->first);
+        default_topics << model.max(-1, termIt->first) << " ||| " << termIt->second << " ||| ";
+       copy(strings.begin(), strings.end(),ostream_iterator<std::string>(default_topics, " "));
+        default_topics <<endl;
+      }
     }
   }
 
-- 
cgit v1.2.3