From d0807c45f28a8ebe02d119f2563bb3835a153e0f Mon Sep 17 00:00:00 2001 From: XapaJIaMnu Date: Tue, 23 Dec 2014 15:21:06 +0000 Subject: Fixed crash in probingPT when probability is precisely 0 --- misc/CreateProbingPT.cpp | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) (limited to 'misc/CreateProbingPT.cpp') diff --git a/misc/CreateProbingPT.cpp b/misc/CreateProbingPT.cpp index 3ea369a96..2b0e8cd8a 100644 --- a/misc/CreateProbingPT.cpp +++ b/misc/CreateProbingPT.cpp @@ -5,14 +5,22 @@ int main(int argc, char* argv[]){ - if (argc != 3) { + const char * is_reordering = "false"; + + if (!(argc == 5 || argc == 4)) { // Tell the user how to run the program - std::cerr << "Provided " << argc << " arguments, needed 3." << std::endl; - std::cerr << "Usage: " << argv[0] << " path_to_phrasetable output_dir" << std::endl; + std::cerr << "Provided " << argc << " arguments, needed 4 or 5." << std::endl; + std::cerr << "Usage: " << argv[0] << " path_to_phrasetable output_dir num_scores is_reordering" << std::endl; + std::cerr << "is_reordering should be either true or false, but it is currently a stub feature." << std::endl; + //std::cerr << "Usage: " << argv[0] << " path_to_phrasetable number_of_uniq_lines output_bin_file output_hash_table output_vocab_id" << std::endl; return 1; } - createProbingPT(argv[1], argv[2]); + if (argc == 5) { + is_reordering = argv[4]; + } + + createProbingPT(argv[1], argv[2], argv[3], is_reordering); util::PrintUsage(std::cout); return 0; -- cgit v1.2.3