Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/moses-smt/vowpal_wabbit.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'vowpalwabbit/oaa.cc')
-rw-r--r--vowpalwabbit/oaa.cc7
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;