add unigram to export k mixture model tool
authorPeng Wu <alexepico@gmail.com>
Tue, 7 Jun 2011 06:39:59 +0000 (14:39 +0800)
committerPeng Wu <alexepico@gmail.com>
Tue, 7 Jun 2011 06:39:59 +0000 (14:39 +0800)
utils/training/export_k_mixture_model.cpp

index f2716fb8b03702f03a5adb3dd73e4eed5dbdc77a..87a45776ecefd3b7460d4cd1a0f2a205d0241593 100644 (file)
@@ -33,8 +33,9 @@ bool print_k_mixture_model_magic_header(FILE * output,
         fprintf(stderr, "no magic header in k mixture model.\n");
         exit(ENODATA);
     }
-    fprintf(output, "\\data model \"k mixture model\" count %d N %d\n",
-           magic_header.m_WC, magic_header.m_N);
+    fprintf(output, "\\data model \"k mixture model\" count %d N %d "
+            "total_freq %d\n", magic_header.m_WC, magic_header.m_N,
+            magic_header.m_total_freq);
     return true;
 }
 
@@ -51,7 +52,8 @@ bool print_k_mixture_model_array_headers(FILE * output,
         assert(bigram->get_array_header(*token, array_header));
         char * phrase = taglib_token_to_string(phrase_index, *token);
         if ( phrase )
-            fprintf(output, "\\item %s count %d\n", phrase, array_header.m_WC);
+            fprintf(output, "\\item %s count %d freq %d\n",
+                    phrase, array_header.m_WC, array_header.m_freq);
 
         g_free(phrase);
     }