CLASSWEIGHTS::classweights Struct Reference

Public Member Functions
void	load_string (std::string const &source)
float	get_class_weight (uint32_t klass)

Public Attributes
std::unordered_map< uint32_t, float >	weights

Detailed Description

Definition at line 8 of file classweight.cc.

Member Function Documentation

get_class_weight()

float CLASSWEIGHTS::classweights::get_class_weight ( uint32_t  klass )

inline

Definition at line 36 of file classweight.cc.

References f.

Referenced by CLASSWEIGHTS::predict_or_learn().

  {
    auto got = weights.find(klass);
    if (got == weights.end())
      return 1.0f;
    else
      return got->second;
  }

load_string()

void CLASSWEIGHTS::classweights::load_string ( std::string const &  source )

inline

Definition at line 12 of file classweight.cc.

References THROW.

  {
    std::stringstream ss(source);
    std::string item;
    while (std::getline(ss, item, ','))
    {
      std::stringstream inner_ss(item);
      std::string klass;
      std::string weight;
      std::getline(inner_ss, klass, ':');
      std::getline(inner_ss, weight, ':');

      if (!klass.size() || !weight.size())
      {
        THROW("error: while parsing --classweight " << item);
      }

      int klass_int = std::stoi(klass);
      float weight_double = std::stof(weight);

      weights[klass_int] = weight_double;
    }
  }

Member Data Documentation

weights

std::unordered_map<uint32_t, float> CLASSWEIGHTS::classweights::weights

Definition at line 10 of file classweight.cc.

---

The documentation for this struct was generated from the following file:

• /mnt/c/w/linux/vowpal_wabbit/vowpalwabbit/classweight.cc