summaryrefslogtreecommitdiffstats
path: root/utils/training/gen_ngram.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'utils/training/gen_ngram.cpp')
-rw-r--r--utils/training/gen_ngram.cpp136
1 files changed, 0 insertions, 136 deletions
diff --git a/utils/training/gen_ngram.cpp b/utils/training/gen_ngram.cpp
deleted file mode 100644
index dbce442..0000000
--- a/utils/training/gen_ngram.cpp
+++ /dev/null
@@ -1,136 +0,0 @@
-/*
- * libzhuyin
- * Library to deal with zhuyin.
- *
- * Copyright (C) 2006-2007, 2011 Peng Wu
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
- */
-
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <locale.h>
-#include <glib.h>
-#include "zhuyin_internal.h"
-#include "utils_helper.h"
-
-static gboolean train_pi_gram = TRUE;
-static const gchar * bigram_filename = SYSTEM_BIGRAM;
-
-static GOptionEntry entries[] =
-{
- {"skip-pi-gram-training", 0, G_OPTION_FLAG_REVERSE, G_OPTION_ARG_NONE, &train_pi_gram, "skip pi-gram training", NULL},
- {"bigram-file", 0, 0, G_OPTION_ARG_FILENAME, &bigram_filename, "bi-gram file", NULL},
- {NULL}
-};
-
-int main(int argc, char * argv[]){
- FILE * input = stdin;
-
- setlocale(LC_ALL, "");
-
- GError * error = NULL;
- GOptionContext * context;
-
- context = g_option_context_new("- generate n-gram");
- g_option_context_add_main_entries(context, entries, NULL);
- if (!g_option_context_parse(context, &argc, &argv, &error)) {
- g_print("option parsing failed:%s\n", error->message);
- exit(EINVAL);
- }
-
- SystemTableInfo system_table_info;
-
- bool retval = system_table_info.load(SYSTEM_TABLE_INFO);
- if (!retval) {
- fprintf(stderr, "load table.conf failed.\n");
- exit(ENOENT);
- }
-
- PhraseLargeTable2 phrase_table;
- /* init phrase table */
- MemoryChunk * chunk = new MemoryChunk;
- chunk->load(SYSTEM_PHRASE_INDEX);
- phrase_table.load(chunk);
-
- FacadePhraseIndex phrase_index;
-
- const pinyin_table_info_t * phrase_files =
- system_table_info.get_table_info();
-
- if (!load_phrase_index(phrase_files, &phrase_index))
- exit(ENOENT);
-
- Bigram bigram;
- bigram.attach(bigram_filename, ATTACH_CREATE|ATTACH_READWRITE);
-
- char* linebuf = NULL; size_t size = 0;
- phrase_token_t last_token, cur_token = last_token = 0;
- while( getline(&linebuf, &size, input) ){
- if ( feof(input) )
- break;
-
- if ( '\n' == linebuf[strlen(linebuf) - 1] ) {
- linebuf[strlen(linebuf) - 1] = '\0';
- }
-
- TAGLIB_PARSE_SEGMENTED_LINE(&phrase_index, token, linebuf);
-
- last_token = cur_token;
- cur_token = token;
-
- /* skip null_token in second word. */
- if ( null_token == cur_token )
- continue;
-
- /* training uni-gram */
- phrase_index.add_unigram_frequency(cur_token, 1);
-
- /* skip pi-gram training. */
- if ( null_token == last_token ){
- if ( !train_pi_gram )
- continue;
- last_token = sentence_start;
- }
-
- /* train bi-gram */
- SingleGram * single_gram = NULL;
- bigram.load(last_token, single_gram);
-
- if ( NULL == single_gram ){
- single_gram = new SingleGram;
- }
- guint32 freq, total_freq;
- /* increase freq */
- if (single_gram->get_freq(cur_token, freq))
- assert(single_gram->set_freq(cur_token, freq + 1));
- else
- assert(single_gram->insert_freq(cur_token, 1));
- /* increase total freq */
- single_gram->get_total_freq(total_freq);
- single_gram->set_total_freq(total_freq + 1);
-
- bigram.store(last_token, single_gram);
- delete single_gram;
- }
-
- free(linebuf);
-
- if (!save_phrase_index(phrase_files, &phrase_index))
- exit(ENOENT);
-
- return 0;
-}