From f6db0114debc6928eaaf1b526d8507a5b045ea85 Mon Sep 17 00:00:00 2001 From: Peng Wu Date: Wed, 11 May 2011 13:47:29 +0800 Subject: refine gen k mixture model --- utils/training/gen_k_mixture_model.cpp | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) (limited to 'utils') diff --git a/utils/training/gen_k_mixture_model.cpp b/utils/training/gen_k_mixture_model.cpp index d0b327a..8fc6946 100644 --- a/utils/training/gen_k_mixture_model.cpp +++ b/utils/training/gen_k_mixture_model.cpp @@ -185,12 +185,9 @@ static void train_single_gram_wrapper(gpointer key, gpointer value, KMixtureModelSingleGram * single_gram = NULL; bool exists = g_k_mixture_model->load(token, single_gram); - if ( exists ){ - train_single_gram(token, single_gram, delta); - } else { /* item doesn't exist. */ + if ( !exists ) single_gram = new KMixtureModelSingleGram; - train_single_gram(token, single_gram, delta); - } + train_single_gram(token, single_gram, delta); KMixtureModelMagicHeader magic_header; assert(g_k_mixture_model->get_magic_header(magic_header)); @@ -275,7 +272,7 @@ int main(int argc, char * argv[]){ g_hash_of_document = NULL; ++i; - } + } delete g_phrases; delete g_k_mixture_model; -- cgit