summaryrefslogtreecommitdiff
path: root/decoder/hg_sampler.h
diff options
context:
space:
mode:
Diffstat (limited to 'decoder/hg_sampler.h')
-rw-r--r--decoder/hg_sampler.h7
1 files changed, 7 insertions, 0 deletions
diff --git a/decoder/hg_sampler.h b/decoder/hg_sampler.h
index bf4e1eb0..6ac39a20 100644
--- a/decoder/hg_sampler.h
+++ b/decoder/hg_sampler.h
@@ -3,6 +3,7 @@
#include <vector>
+#include <string>
#include "sparse_vector.h"
#include "sampler.h"
#include "wordid.h"
@@ -22,6 +23,12 @@ struct HypergraphSampler {
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