diff options
Diffstat (limited to 'vowpalwabbit/cb_algs.cc')
-rw-r--r-- | vowpalwabbit/cb_algs.cc | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/vowpalwabbit/cb_algs.cc b/vowpalwabbit/cb_algs.cc index a9827f27..2d8a2252 100644 --- a/vowpalwabbit/cb_algs.cc +++ b/vowpalwabbit/cb_algs.cc @@ -497,10 +497,14 @@ namespace CB_ALGS } if (!vm.count("csoaa")) - vm.insert(pair<string,po::variable_value>(string("csoaa"),vm["cb"])); + { + all.args.push_back("--csoaa"); + stringstream ss; + ss << vm["cb"].as<size_t>(); + all.args.push_back(ss.str()); + } base_learner* base = setup_base(all,vm); - if (eval) all.p->lp = CB_EVAL::cb_eval; else |