diff options
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 |