/*
Copyright (c) by respective owners including Yahoo!, Microsoft, and
individual contributors. All rights reserved.  Released under a BSD
license as described in the file LICENSE.
 */
#pragma once

#include "label_parser.h"
#include <vector>

struct example;

namespace CB
{
struct cb_class
{
  float cost;         // the cost of this class
  uint32_t action;    // the index of this class
  float probability;  // new for bandit setting, specifies the probability the data collection policy chose this class
                      // for importance weighting
  float partial_prediction;  // essentially a return value
  bool operator==(cb_class j) { return action == j.action; }
};

struct label
{
  v_array<cb_class> costs;
  float weight;
};

extern label_parser cb_label;                  // for learning
bool ec_is_example_header(example const& ec);  // example headers look like "shared"

void print_update(vw& all, bool is_test, example& ec, std::vector<example*>* ec_seq, bool action_scores);
}  // namespace CB

namespace CB_EVAL
{
struct label
{
  uint32_t action;
  CB::label event;
};

extern label_parser cb_eval;  // for evaluation of an arbitrary policy.
}  // namespace CB_EVAL