topk.cc File Reference

#include <cfloat>
#include <sstream>
#include <queue>
#include <utility>
#include "topk.h"
#include "learner.h"
#include "parse_args.h"
#include "vw.h"

Go to the source code of this file.

Classes
class	VW::topk

Namespaces
	VW

Functions
void	print_result (int file_descriptor, std::pair< VW::topk::const_iterator_t, VW::topk::const_iterator_t > const &view)
void	output_example (vw &all, example &ec)
template<bool is_learn>
void	predict_or_learn (VW::topk &d, LEARNER::single_learner &base, multi_ex &ec_seq)
void	finish_example (vw &all, VW::topk &d, multi_ex &ec_seq)
LEARNER::base_learner *	topk_setup (options_i &options, vw &all)

Function Documentation

finish_example()

void finish_example	(	vw &	all,
		VW::topk &	d,
		multi_ex &	ec_seq
	)

Definition at line 124 of file topk.cc.

References VW::topk::clear_container(), vw::final_prediction_sink, VW::finish_example(), VW::topk::get_container_view(), output_example(), and print_result().

{
  for (auto ec : ec_seq) output_example(all, *ec);
  for (auto sink : all.final_prediction_sink) print_result(sink, d.get_container_view());
  d.clear_container();
  VW::finish_example(all, ec_seq);
}

output_example()

void output_example	(	vw &	all,
		example &	ec
	)

Definition at line 104 of file topk.cc.

References example::l, label_data::label, example::loss, example::num_features, CB::print_update(), vw::sd, polylabel::simple, example::test_only, shared_data::update(), example::weight, and shared_data::weighted_labels.

Referenced by finish_example().

{
  label_data& ld = ec.l.simple;

  all.sd->update(ec.test_only, ld.label != FLT_MAX, ec.loss, ec.weight, ec.num_features);
  if (ld.label != FLT_MAX)
    all.sd->weighted_labels += ((double)ld.label) * ec.weight;

  print_update(all, ec);
}

predict_or_learn()

template<bool is_learn>

void predict_or_learn	(	VW::topk &	d,
		LEARNER::single_learner &	base,
		multi_ex &	ec_seq
	)

Definition at line 116 of file topk.cc.

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

{
  if (is_learn)
    d.learn(base, ec_seq);
  else
    d.predict(base, ec_seq);
}

print_result()

void print_result	(	int	file_descriptor,
		std::pair< VW::topk::const_iterator_t, VW::topk::const_iterator_t > const &	view
	)

Definition at line 81 of file topk.cc.

References print_tag().

Referenced by finish_example().

{
  if (file_descriptor >= 0)
  {
    std::stringstream ss;
    for (auto it = view.first; it != view.second; it++)
    {
      ss << std::fixed << it->first << " ";
      print_tag(ss, it->second);
      ss << " \n";
    }
    ss << '\n';
    ssize_t len = ss.str().size();
#ifdef _WIN32
    ssize_t t = _write(file_descriptor, ss.str().c_str(), (unsigned int)len);
#else
    ssize_t t = write(file_descriptor, ss.str().c_str(), (unsigned int)len);
#endif
    if (t != len)
      std::cerr << "write error: " << strerror(errno) << std::endl;
  }
}

topk_setup()

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

Definition at line 132 of file topk.cc.

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

Referenced by parse_reductions().

{
  uint32_t K;
  option_group_definition new_options("Top K");
  new_options.add(make_option("top", K).keep().help("top k recommendation"));
  options.add_and_parse(new_options);

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

  auto data = scoped_calloc_or_throw<VW::topk>(K);

  LEARNER::learner<VW::topk, multi_ex>& l =
      init_learner(data, as_singleline(setup_base(options, all)), predict_or_learn<true>, predict_or_learn<false>);
  l.set_finish_example(finish_example);

  return make_base(l);
}