autolink.cc File Reference

#include "autolink.h"
#include "learner.h"
#include "global_data.h"
#include "parse_args.h"
#include <cstdint>

Go to the source code of this file.

Classes
struct	VW::autolink

Namespaces
	VW

Functions
template<bool is_learn>
void	predict_or_learn (VW::autolink &b, LEARNER::single_learner &base, example &ec)
LEARNER::base_learner *	autolink_setup (options_i &options, vw &all)

Function Documentation

autolink_setup()

LEARNER::base_learner* autolink_setup	(	options_i &	options,
		vw &	all
	)

Definition at line 85 of file autolink.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(), setup_base(), parameters::stride_shift(), VW::config::options_i::was_supplied(), and vw::weights.

Referenced by parse_reductions().

{
  uint32_t d;
  option_group_definition new_options("Autolink");
  new_options.add(make_option("autolink", d).keep().help("create link function with polynomial d"));
  options.add_and_parse(new_options);

  if (!options.was_supplied("autolink"))
    return nullptr;

  auto autolink_reduction = scoped_calloc_or_throw<VW::autolink>(d, all.weights.stride_shift());
  return make_base(init_learner(
      autolink_reduction, as_singleline(setup_base(options, all)), predict_or_learn<true>, predict_or_learn<false>));
}

predict_or_learn()

template<bool is_learn>

void predict_or_learn	(	VW::autolink &	b,
		LEARNER::single_learner &	base,
		example &	ec
	)

Definition at line 77 of file autolink.cc.

References VW::autolink::learn(), and VW::autolink::predict().

{
  if (is_learn)
    b.learn(base, ec);
  else
    b.predict(base, ec);
}