From c5ec52ded3f14271e25e97cefc8bac03b176f297 Mon Sep 17 00:00:00 2001
From: Chris Dyer <cdyer@cs.cmu.edu>
Date: Mon, 16 Apr 2012 00:18:20 -0400
Subject: rst algorithm

---
 rst_parser/rst.h | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'rst_parser/rst.h')

diff --git a/rst_parser/rst.h b/rst_parser/rst.h
index 865871eb..a269ff9b 100644
--- a/rst_parser/rst.h
+++ b/rst_parser/rst.h
@@ -1,10 +1,15 @@
 #ifndef _RST_H_
 #define _RST_H_
 
+#include <vector>
+#include "sampler.h"
 #include "arc_factored.h"
 
-struct StochasticForest {
-  explicit StochasticForest(const ArcFactoredForest& af);
+struct TreeSampler {
+  explicit TreeSampler(const ArcFactoredForest& af);
+  void SampleRandomSpanningTree(EdgeSubset* tree);
+  const ArcFactoredForest& forest;
+  std::vector<SampleSet<double> > usucc;
 };
 
 #endif
-- 
cgit v1.2.3