Vowpal Wabbit
Functions
binary.h File Reference

Go to the source code of this file.

Functions

LEARNER::base_learnerbinary_setup (VW::config::options_i &options, vw &all)
 

Function Documentation

◆ binary_setup()

LEARNER::base_learner* binary_setup ( VW::config::options_i options,
vw all 
)

Definition at line 30 of file binary.cc.

References VW::config::option_group_definition::add(), VW::config::options_i::add_and_parse(), LEARNER::as_singleline(), LEARNER::init_learner(), LEARNER::make_base(), VW::config::make_option(), and setup_base().

Referenced by parse_reductions().

31 {
32  bool binary = false;
33  option_group_definition new_options("Binary loss");
34  new_options.add(make_option("binary", binary).keep().help("report loss as binary classification on -1,1"));
35  options.add_and_parse(new_options);
36 
37  if (!binary)
38  return nullptr;
39 
41  LEARNER::init_learner(as_singleline(setup_base(options, all)), predict_or_learn<true>, predict_or_learn<false>);
42  return make_base(ret);
43 }
base_learner * make_base(learner< T, E > &base)
Definition: learner.h:462
virtual void add_and_parse(const option_group_definition &group)=0
single_learner * as_singleline(learner< T, E > *l)
Definition: learner.h:476
learner< T, E > & init_learner(free_ptr< T > &dat, L *base, void(*learn)(T &, L &, E &), void(*predict)(T &, L &, E &), size_t ws, prediction_type::prediction_type_t pred_type)
Definition: learner.h:369
typed_option< T > make_option(std::string name, T &location)
Definition: options.h:80
LEARNER::base_learner * setup_base(options_i &options, vw &all)
Definition: parse_args.cc:1222