decoder/hg_sampler.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

#ifndef HG_SAMPLER_H_
#define HG_SAMPLER_H_

#include <vector>
#include <string>
#include "sparse_vector.h"
#include "sampler.h"
#include "wordid.h"

class Hypergraph;

struct HypergraphSampler {

  struct Hypothesis {
    std::vector<WordID> words;
    SparseVector<double> fmap;
    prob_t model_score;   // log unnormalized probability
  };

  static void
  sample_hypotheses(const Hypergraph& hg,
                    unsigned n,   // how many samples to draw
                    MT19937* rng,
                    std::vector<Hypothesis>* hypos);

  static void
  sample_trees(const Hypergraph& hg,
               unsigned n,
               MT19937* rng,
               std::vector<std::string>* trees);
};

#endif