diff options
Diffstat (limited to 'vowpalwabbit/csoaa.cc')
-rw-r--r-- | vowpalwabbit/csoaa.cc | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/vowpalwabbit/csoaa.cc b/vowpalwabbit/csoaa.cc index a5429dd0..0928735a 100644 --- a/vowpalwabbit/csoaa.cc +++ b/vowpalwabbit/csoaa.cc @@ -86,9 +86,8 @@ namespace CSOAA { all.p->lp = cs_label; all.sd->k = nb_actions; - learner<csoaa>& l = init_learner(&c, all.l, nb_actions); - l.set_learn(predict_or_learn<true>); - l.set_predict(predict_or_learn<false>); + learner<csoaa>& l = init_learner(&c, all.l, predict_or_learn<true>, + predict_or_learn<false>, nb_actions); l.set_finish_example(finish_example); return make_base(l); } @@ -715,9 +714,7 @@ namespace LabelDict { ld.read_example_this_loop = 0; ld.need_to_clear = false; - learner<ldf>& l = init_learner(&ld, all.l); - l.set_learn(predict_or_learn<true>); - l.set_predict(predict_or_learn<false>); + learner<ldf>& l = init_learner(&ld, all.l, predict_or_learn<true>, predict_or_learn<false>); if (ld.is_singleline) l.set_finish_example(finish_singleline_example); else |