diff options
Diffstat (limited to 'utils/training/gen_unigram.cpp')
-rw-r--r-- | utils/training/gen_unigram.cpp | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/utils/training/gen_unigram.cpp b/utils/training/gen_unigram.cpp index 022f159..cd938f6 100644 --- a/utils/training/gen_unigram.cpp +++ b/utils/training/gen_unigram.cpp @@ -31,12 +31,20 @@ int main(int argc, char * argv[]){ /* gb_char binary file */ MemoryChunk * chunk = new MemoryChunk; - chunk->load("gb_char.bin"); + bool retval = chunk->load("gb_char.bin"); + if (!retval) { + fprintf(stderr, "open gb_char.bin failed!\n"); + exit(ENOENT); + } phrase_index.load(1, chunk); /* gbk_char binary file */ chunk = new MemoryChunk; - chunk->load("gbk_char.bin"); + retval = chunk->load("gbk_char.bin"); + if (!retval) { + fprintf(stderr, "open gbk_char.bin failed!\n"); + exit(ENOENT); + } phrase_index.load(2, chunk); /* Note: please increase the value when corpus size becomes larger. |