score_map.h

Go to the documentation of this file.

#ifndef __ZGEN_SHIFTREDUCE_MODEL_SCORE_MAP_H__
#define __ZGEN_SHIFTREDUCE_MODEL_SCORE_MAP_H__

#include "utils/serialization/unordered_map.h"
#include "shiftreduce/types/action.h"
#include "shiftreduce/model/score_context.h"
#include "ml/perceptron/param.h"

namespace ZGen {
namespace ShiftReduce {

template<class ScoreType>
class ScoreMap {
private:
  typedef MachineLearning::Perceptron::Parameter param_t;
  typedef boost::unordered_map<ScoreType, param_t> map_t;
  typedef std::vector<ScoreType> cache_t;
  typedef std::function<void(const ScoreContext&, const Action&, cache_t&)> extractor_t;
  //typedef std::function<void(const ScoreContext&, const Action&)> extractor_t;
public:
  ScoreMap(extractor_t _extractor): extractor(_extractor) {}

  floatval_t score(const ScoreContext& ctx, const Action& act,
      bool avg, floatval_t default_return_value = 0.) {
    cache.clear();
    extractor(ctx, act, cache);
    //extractor(ctx, act);
    floatval_t ret = 0.;
    for (int i = 0; i < cache.size(); ++ i) {
      const ScoreType& entry = cache[i];
      typename map_t::const_iterator itx = payload.find(entry);
      if (itx != payload.end()) {
        if (avg) {
          ret += itx->second.w_sum;
        } else {
          ret += itx->second.w;
        }
      }
    }
    return ret;
  }

  void update(const ScoreContext& ctx, const Action& act,
      int now, floatval_t scale = 1.) {
    cache.clear();
    extractor(ctx, act, cache);
    for (int i = 0; i < cache.size(); ++ i) {
      const ScoreType& entry = cache[i];
      param_t& param = payload[entry];
      param.add(now, scale);
    }
  }

  void flush(int now) {
    for (typename map_t::iterator itx = payload.begin();
        itx != payload.end(); ++ itx) {
      itx->second.flush(now);
    }
  }

  void save(boost::archive::text_oarchive& oa) {
    oa << payload;
  }

  void load(boost::archive::text_iarchive& ia) {
    ia >> payload;
  }

private:
  map_t payload;
  cache_t cache;
  extractor_t extractor;
};


}
}

#endif  //  end for __ZGEN_SHIFTREDUCE_MODEL_SCORE_MAP_H__