diff options
-rw-r--r-- | vowpalwabbit/lrq.cc | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/vowpalwabbit/lrq.cc b/vowpalwabbit/lrq.cc index 13534375..d035f1f4 100644 --- a/vowpalwabbit/lrq.cc +++ b/vowpalwabbit/lrq.cc @@ -207,13 +207,14 @@ namespace LRQ { if (all.vm.count("random_seed")) random_seed = all.vm["random_seed"].as<size_t> (); lrq.initial_seed = lrq.seed = random_seed | 8675309; - if (all.vm.count("lrqdropout")) - lrq.dropout = true; + if (all.vm.count("lrqdropout")) + { + lrq.dropout = true; + *all.file_options << " --lrqdropout "; + } else lrq.dropout = false; - *all.file_options << " --lrqdropout "; - lrq.lrpairs = all.vm["lrq"].as<vector<string> > (); for (vector<string>::iterator i = lrq.lrpairs.begin (); |