diff options
Diffstat (limited to 'vowpalwabbit/stagewise_poly.cc')
-rw-r--r-- | vowpalwabbit/stagewise_poly.cc | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/vowpalwabbit/stagewise_poly.cc b/vowpalwabbit/stagewise_poly.cc index fe4f6fbd..c435f45b 100644 --- a/vowpalwabbit/stagewise_poly.cc +++ b/vowpalwabbit/stagewise_poly.cc @@ -699,9 +699,7 @@ namespace StagewisePoly //following is so that saved models know to load us. *all.file_options << " --stage_poly"; - learner<stagewise_poly>& l = init_learner(&poly, all.l); - l.set_learn(learn); - l.set_predict(predict); + learner<stagewise_poly>& l = init_learner(&poly, all.l, learn, predict); l.set_finish(finish); l.set_save_load(save_load); l.set_finish_example(finish_example); |