cb_sample.cc File Reference

#include "reductions.h"
#include "cb_sample.h"
#include "explore.h"
#include "rand48.h"

Go to the source code of this file.

Classes
struct	VW::cb_sample_data

Namespaces
	VW

Functions
template<bool is_learn>
void	learn_or_predict (cb_sample_data &data, multi_learner &base, multi_ex &examples)
base_learner *	cb_sample_setup (options_i &options, vw &all)

Function Documentation

cb_sample_setup()

base_learner* cb_sample_setup	(	options_i &	options,
		vw &	all
	)

Definition at line 97 of file cb_sample.cc.

References prediction_type::action_probs, VW::config::option_group_definition::add(), VW::config::options_i::add_and_parse(), LEARNER::as_multiline(), vw::get_random_state(), LEARNER::init_learner(), LEARNER::make_base(), VW::config::make_option(), setup_base(), THROW, and VW::config::options_i::was_supplied().

Referenced by parse_reductions().

{
  bool cb_sample_option = false;

  option_group_definition new_options("CB Sample");
  new_options.add(make_option("cb_sample", cb_sample_option).keep().help("Sample from CB pdf and swap top action."));
  options.add_and_parse(new_options);

  if (!cb_sample_option)
    return nullptr;

  if (options.was_supplied("no_predict"))
  {
    THROW("cb_sample cannot be used with no_predict, as there would be no predictions to sample.");
  }

  auto data = scoped_calloc_or_throw<cb_sample_data>(all.get_random_state());
  return make_base(init_learner(data, as_multiline(setup_base(options, all)), learn_or_predict<true>,
      learn_or_predict<false>, 1 /* weights */, prediction_type::action_probs));
}

learn_or_predict()

template<bool is_learn>

void learn_or_predict	(	cb_sample_data &	data,
		multi_learner &	base,
		multi_ex &	examples
	)

Definition at line 92 of file cb_sample.cc.

References VW::cb_sample_data::learn_or_predict().

{
  data.learn_or_predict<is_learn>(base, examples);
}