From 594cc714737708bb7c90c24a1ab1537b052f45ee Mon Sep 17 00:00:00 2001 From: redpony Date: Sun, 29 Aug 2010 00:36:09 +0000 Subject: online optimizer git-svn-id: https://ws10smt.googlecode.com/svn/trunk@631 ec762483-ff6d-05da-a07a-a48fb63a330f --- training/mr_optimize_reduce.cc | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) (limited to 'training/mr_optimize_reduce.cc') diff --git a/training/mr_optimize_reduce.cc b/training/mr_optimize_reduce.cc index 42727ecb..b931991d 100644 --- a/training/mr_optimize_reduce.cc +++ b/training/mr_optimize_reduce.cc @@ -108,11 +108,9 @@ int main(int argc, char** argv) { } wm.InitVector(&means); } - shared_ptr o; + shared_ptr o; const string omethod = conf["optimization_method"].as(); - if (omethod == "sgd") - o.reset(new SGDOptimizer(conf["eta"].as())); - else if (omethod == "rprop") + if (omethod == "rprop") o.reset(new RPropOptimizer(num_feats)); // TODO add configuration else o.reset(new LBFGSOptimizer(num_feats, conf["correction_buffers"].as())); -- cgit v1.2.3