|
|
|
|
|
#include "SearchOptions.h" |
|
|
#include "../legacy/Parameter.h" |
|
|
|
|
|
namespace Moses2 |
|
|
{ |
|
|
|
|
|
SearchOptions:: |
|
|
SearchOptions() |
|
|
: algo(Normal) |
|
|
, stack_size(DEFAULT_MAX_HYPOSTACK_SIZE) |
|
|
, stack_diversity(0) |
|
|
, disable_discarding(false) |
|
|
, max_phrase_length(DEFAULT_MAX_PHRASE_LENGTH) |
|
|
, max_trans_opt_per_cov(DEFAULT_MAX_TRANS_OPT_SIZE) |
|
|
, max_partial_trans_opt(DEFAULT_MAX_PART_TRANS_OPT_SIZE) |
|
|
, beam_width(DEFAULT_BEAM_WIDTH) |
|
|
, timeout(0) |
|
|
, consensus(false) |
|
|
, early_discarding_threshold(DEFAULT_EARLY_DISCARDING_THRESHOLD) |
|
|
, trans_opt_threshold(DEFAULT_TRANSLATION_OPTION_THRESHOLD) |
|
|
{ } |
|
|
|
|
|
SearchOptions:: |
|
|
SearchOptions(Parameter const& param) |
|
|
: stack_diversity(0) |
|
|
{ |
|
|
init(param); |
|
|
} |
|
|
|
|
|
bool |
|
|
SearchOptions:: |
|
|
init(Parameter const& param) |
|
|
{ |
|
|
param.SetParameter(algo, "search-algorithm", Normal); |
|
|
param.SetParameter(stack_size, "stack", DEFAULT_MAX_HYPOSTACK_SIZE); |
|
|
param.SetParameter(stack_diversity, "stack-diversity", size_t(0)); |
|
|
param.SetParameter(beam_width, "beam-threshold", DEFAULT_BEAM_WIDTH); |
|
|
param.SetParameter(early_discarding_threshold, "early-discarding-threshold", |
|
|
DEFAULT_EARLY_DISCARDING_THRESHOLD); |
|
|
param.SetParameter(timeout, "time-out", 0); |
|
|
param.SetParameter(max_phrase_length, "max-phrase-length", |
|
|
DEFAULT_MAX_PHRASE_LENGTH); |
|
|
param.SetParameter(trans_opt_threshold, "translation-option-threshold", |
|
|
DEFAULT_TRANSLATION_OPTION_THRESHOLD); |
|
|
param.SetParameter(max_trans_opt_per_cov, "max-trans-opt-per-coverage", |
|
|
DEFAULT_MAX_TRANS_OPT_SIZE); |
|
|
param.SetParameter(max_partial_trans_opt, "max-partial-trans-opt", |
|
|
DEFAULT_MAX_PART_TRANS_OPT_SIZE); |
|
|
|
|
|
param.SetParameter(consensus, "consensus-decoding", false); |
|
|
param.SetParameter(disable_discarding, "disable-discarding", false); |
|
|
|
|
|
|
|
|
beam_width = TransformScore(beam_width); |
|
|
trans_opt_threshold = TransformScore(trans_opt_threshold); |
|
|
early_discarding_threshold = TransformScore(early_discarding_threshold); |
|
|
|
|
|
return true; |
|
|
} |
|
|
|
|
|
bool |
|
|
is_syntax(SearchAlgorithm algo) |
|
|
{ |
|
|
return (algo == CYKPlus || algo == ChartIncremental || |
|
|
algo == SyntaxS2T || algo == SyntaxT2S || |
|
|
algo == SyntaxF2S || algo == SyntaxT2S_SCFG); |
|
|
} |
|
|
|
|
|
#ifdef HAVE_XMLRPC_C |
|
|
bool |
|
|
SearchOptions:: |
|
|
update(std::map<std::string,xmlrpc_c::value>const& params) |
|
|
{ |
|
|
typedef std::map<std::string, xmlrpc_c::value> params_t; |
|
|
|
|
|
params_t::const_iterator si = params.find("search-algorithm"); |
|
|
if (si != params.end()) { |
|
|
|
|
|
std::string spec = xmlrpc_c::value_string(si->second); |
|
|
if (spec == "normal" || spec == "0") algo = Normal; |
|
|
else if (spec == "cube" || spec == "1") algo = CubePruning; |
|
|
else throw xmlrpc_c::fault("Unsupported search algorithm", |
|
|
xmlrpc_c::fault::CODE_PARSE); |
|
|
} |
|
|
|
|
|
si = params.find("stack"); |
|
|
if (si != params.end()) stack_size = xmlrpc_c::value_int(si->second); |
|
|
|
|
|
si = params.find("stack-diversity"); |
|
|
if (si != params.end()) stack_diversity = xmlrpc_c::value_int(si->second); |
|
|
|
|
|
si = params.find("beam-threshold"); |
|
|
if (si != params.end()) beam_width = xmlrpc_c::value_double(si->second); |
|
|
|
|
|
si = params.find("time-out"); |
|
|
if (si != params.end()) timeout = xmlrpc_c::value_int(si->second); |
|
|
|
|
|
si = params.find("max-phrase-length"); |
|
|
if (si != params.end()) max_phrase_length = xmlrpc_c::value_int(si->second); |
|
|
|
|
|
return true; |
|
|
} |
|
|
#endif |
|
|
|
|
|
} |
|
|
|
