blob: 8bf389f7a3d24a10ff699e067ce70e142f0ed7d2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
#ifndef _RST_H_
#define _RST_H_
#include <vector>
#include "sampler.h"
#include "arc_factored.h"
#include "alias_sampler.h"
struct TreeSampler {
explicit TreeSampler(const ArcFactoredForest& af);
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
|
