diff options
author | Chris Dyer <cdyer@cs.cmu.edu> | 2012-04-16 18:20:33 -0400 |
---|---|---|
committer | Chris Dyer <cdyer@cs.cmu.edu> | 2012-04-16 18:20:33 -0400 |
commit | 4b38556c88c739de82b9c298261a262ec620280e (patch) | |
tree | 272124012fa38a359a7c0efb1da4499a1b4371b5 /rst_parser/rst.h | |
parent | f2fcf9e8aa0e5dee75fd08ee915488ec1a741975 (diff) |
rst sampler
Diffstat (limited to 'rst_parser/rst.h')
-rw-r--r-- | rst_parser/rst.h | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/rst_parser/rst.h b/rst_parser/rst.h index a269ff9b..8bf389f7 100644 --- a/rst_parser/rst.h +++ b/rst_parser/rst.h @@ -4,12 +4,18 @@ #include <vector> #include "sampler.h" #include "arc_factored.h" +#include "alias_sampler.h" struct TreeSampler { explicit TreeSampler(const ArcFactoredForest& af); - void SampleRandomSpanningTree(EdgeSubset* tree); + void SampleRandomSpanningTree(EdgeSubset* tree, MT19937* rng); const ArcFactoredForest& forest; +#define USE_ALIAS_SAMPLER 1 +#if USE_ALIAS_SAMPLER + std::vector<AliasSampler> usucc; +#else std::vector<SampleSet<double> > usucc; +#endif }; #endif |