diff options
Diffstat (limited to 'vowpalwabbit/oaa.cc')
-rw-r--r-- | vowpalwabbit/oaa.cc | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/vowpalwabbit/oaa.cc b/vowpalwabbit/oaa.cc index 2328b00d..03a7cac6 100644 --- a/vowpalwabbit/oaa.cc +++ b/vowpalwabbit/oaa.cc @@ -64,6 +64,13 @@ namespace OAA { LEARNER::base_learner* setup(vw& all, po::variables_map& vm) { + po::options_description opts("One-against-all options"); + opts.add_options() + ("oaa", po::value<size_t>(), "Use one-against-all multiclass learning with <k> labels"); + vm = add_options(all, opts); + if(!vm.count("oaa")) + return NULL; + oaa& data = calloc_or_die<oaa>(); data.k = vm["oaa"].as<size_t>(); data.shouldOutput = all.raw_prediction > 0; |