diff options
Diffstat (limited to 'utils/training/gen_ngram.cpp')
-rw-r--r-- | utils/training/gen_ngram.cpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/utils/training/gen_ngram.cpp b/utils/training/gen_ngram.cpp index 918c9c8..367728a 100644 --- a/utils/training/gen_ngram.cpp +++ b/utils/training/gen_ngram.cpp @@ -35,12 +35,12 @@ void print_help(){ } int main(int argc, char * argv[]){ - int i = 1; + int i = 1; bool train_pi_gram = true; bool train_unigram = true; const char * bigram_filename = "../../data/bigram.db"; - setlocale(LC_ALL,""); + setlocale(LC_ALL, ""); while ( i < argc ){ if ( strcmp("--help", argv[i] ) == 0){ print_help(); @@ -52,7 +52,9 @@ int main(int argc, char * argv[]){ if ( ++i >= argc ) print_help(); bigram_filename = argv[i]; - } + }else{ + print_help(); + } ++i; } |