add unigram to merge k mixture model

diff --git a/utils/training/merge_k_mixture_model.cpp b/utils/training/merge_k_mixture_model.cpp
index 13db0eff843a5751270e1cce0d118a19980dc083..cadc0a6877ac4bf018fa0bde57767f21716b8ea9 100644 (file)
--- a/utils/training/merge_k_mixture_model.cpp
+++ b/utils/training/merge_k_mixture_model.cpp
@@ -104,11 +104,19 @@ static bool merge_magic_header( /* in & out */ KMixtureModelBigram * target,
         fprintf(stderr, "the m_WC integer in magic header overflows.\n");
         return false;
     }
+    if ( target_magic_header.m_total_freq + new_magic_header.m_total_freq <
+         std_lite::max( target_magic_header.m_total_freq,
+                        new_magic_header.m_total_freq ) ){
+        fprintf(stderr, "the m_total_freq in magic header overflows.\n");
+        return false;
+    }
 
     merged_magic_header.m_WC = target_magic_header.m_WC +
         new_magic_header.m_WC;
     merged_magic_header.m_N = target_magic_header.m_N +
         new_magic_header.m_N;
+    merged_magic_header.m_total_freq = target_magic_header.m_total_freq +
+        new_magic_header.m_total_freq;
 
     assert(target->set_magic_header(merged_magic_header));
     return true;
@@ -144,6 +152,8 @@ static bool merge_array_items( /* in & out */ KMixtureModelBigram * target,
 
         merged_array_header.m_WC = target_array_header.m_WC +
             new_array_header.m_WC;
+        merged_array_header.m_freq = target_array_header.m_freq +
+            new_array_header.m_freq;
         /* end of word count in array header computing. */
 
         assert(NULL != target_single_gram);