From 9b2da83a560931baab56b73214793b8b664a2689 Mon Sep 17 00:00:00 2001 From: graehl Date: Sun, 18 Jul 2010 23:33:03 +0000 Subject: boost cmd line notify(conf) was missing git-svn-id: https://ws10smt.googlecode.com/svn/trunk@313 ec762483-ff6d-05da-a07a-a48fb63a330f --- decoder/cdec.cc | 11 +++++++---- decoder/ff_bleu.cc | 4 ++-- decoder/oracle_bleu.h | 37 +++++++++++++++++++++++++------------ 3 files changed, 34 insertions(+), 18 deletions(-) (limited to 'decoder') diff --git a/decoder/cdec.cc b/decoder/cdec.cc index 8827cce3..a9c1cb3b 100644 --- a/decoder/cdec.cc +++ b/decoder/cdec.cc @@ -91,7 +91,7 @@ void print_options(std::ostream &out,po::options_description const& opts) { } -void InitCommandLine(int argc, char** argv, po::variables_map* confp) { +void InitCommandLine(int argc, char** argv, OracleBleu &ob, po::variables_map* confp) { po::variables_map &conf=*confp; po::options_description opts("Configuration options"); opts.add_options() @@ -151,7 +151,7 @@ void InitCommandLine(int argc, char** argv, po::variables_map* confp) { ("combine_size,C",po::value()->default_value(1), "When option -G is used, process this many sentence pairs before writing the gradient (1=emit after every sentence pair)") ("forest_output,O",po::value(),"Directory to write forests to") ("minimal_forests,m","Write minimal forests (excludes Rule information). Such forests can be used for ML/MAP training, but not rescoring, etc."); - OracleBleu::AddOptions(&opts); + ob.AddOptions(&opts); po::options_description clo("Command line options"); clo.add_options() ("config,c", po::value(), "Configuration file") @@ -206,6 +206,8 @@ void InitCommandLine(int argc, char** argv, po::variables_map* confp) { cerr << dcmdline_options << endl; exit(1); } + po::notify(conf); + } // TODO move out of cdec into some sampling decoder file @@ -358,7 +360,9 @@ int main(int argc, char** argv) { global_ff_registry.reset(new FFRegistry); register_feature_functions(); po::variables_map conf; - InitCommandLine(argc, argv, &conf); + OracleBleu oracle; + + InitCommandLine(argc, argv, oracle, &conf); const bool write_gradient = conf.count("cll_gradient"); const bool feature_expectations = conf.count("feature_expectations"); if (write_gradient && feature_expectations) { @@ -488,7 +492,6 @@ int main(int argc, char** argv) { const bool crf_uniform_empirical = conf.count("crf_uniform_empirical"); const bool get_oracle_forest = conf.count("get_oracle_forest"); - OracleBleu oracle; if (get_oracle_forest) oracle.UseConf(conf); diff --git a/decoder/ff_bleu.cc b/decoder/ff_bleu.cc index ab61ed10..f8d62aa2 100644 --- a/decoder/ff_bleu.cc +++ b/decoder/ff_bleu.cc @@ -243,8 +243,8 @@ BLEUModel::BLEUModel(const string& param) : //loop over argv and load all references into vector of NgramMaps if (argc >= 1) { - if (argv[1] != "-o" || argc<2) { - cerr<(argv[1]); diff --git a/decoder/oracle_bleu.h b/decoder/oracle_bleu.h index cc19fbca..07d83b7f 100755 --- a/decoder/oracle_bleu.h +++ b/decoder/oracle_bleu.h @@ -73,13 +73,13 @@ struct OracleBleu { WeightVector feature_weights_; DocScorer ds; - static void AddOptions(boost::program_options::options_description *opts) { + void AddOptions(boost::program_options::options_description *opts) { using namespace boost::program_options; using namespace std; opts->add_options() - ("references,R", value(), "Translation reference files") - ("oracle_loss", value(), "IBM_BLEU_3 (default), IBM_BLEU etc") - ("bleu_weight", value()->default_value(1.), "weight to give the hope/fear loss function vs. model score") + ("references,R", value(&refs), "Translation reference files") + ("oracle_loss", value(&loss_name)->default_value("IBM_BLEU_3"), "IBM_BLEU_3 (default), IBM_BLEU etc") + ("bleu_weight", value(&bleu_weight)->default_value(1.), "weight to give the hope/fear loss function vs. model score") ; } int order; @@ -110,28 +110,41 @@ struct OracleBleu { } double bleu_weight; - void UseConf(boost::program_options::variables_map const& conf) { + // you have to call notify(conf) yourself, once, in main or similar + void UseConf(boost::program_options::variables_map const& /* conf */) { using namespace std; - bleu_weight=conf["bleu_weight"].as(); - set_loss(conf["oracle_loss"].as()); - set_refs(conf["references"].as()); + // bleu_weight=conf["bleu_weight"].as(); + //set_loss(conf["oracle_loss"].as()); + //set_refs(conf["references"].as()); + init_loss(); + init_refs(); } ScoreType loss; -// std::string loss_name; + std::string loss_name; boost::shared_ptr pff; - void set_loss(std::string const& lossd="IBM_BLEU_3") { -// loss_name=lossd; - loss=ScoreTypeFromString(lossd); + void set_loss(std::string const& lossd) { + loss_name=lossd; + init_loss(); + } + void init_loss() { + loss=ScoreTypeFromString(loss_name); order=(loss==IBM_BLEU_3)?3:4; std::ostringstream param; param<<"-o "<Create("BLEUModel",param.str()); } + bool is_null() const { + return refs.empty(); + } void set_refs(Refs const& r) { refs=r; + init_refs(); + } + void init_refs() { + if (is_null()) return; assert(refs.size()); ds=DocScorer(loss,refs); doc_score.reset(); -- cgit v1.2.3