diff options
Diffstat (limited to 'vowpalwabbit/nn.cc')
-rw-r--r-- | vowpalwabbit/nn.cc | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/vowpalwabbit/nn.cc b/vowpalwabbit/nn.cc index a7079c95..44876e67 100644 --- a/vowpalwabbit/nn.cc +++ b/vowpalwabbit/nn.cc @@ -365,8 +365,10 @@ CONVERSE: // That's right, I'm using goto. So sue me. n.xsubi = vm["random_seed"].as<size_t>(); n.save_xsubi = n.xsubi; - n.increment = all.l->increment;//Indexing of output layer is odd. - learner<nn>& l = init_learner(&n, setup_base(all,vm), predict_or_learn<true>, + + base_learner* base = setup_base(all,vm); + n.increment = base->increment;//Indexing of output layer is odd. + learner<nn>& l = init_learner(&n, base, predict_or_learn<true>, predict_or_learn<false>, n.k+1); l.set_finish(finish); l.set_finish_example(finish_example); |