kbest repo

1 files changed, 19 insertions, 0 deletions

diff --git a/training/kbest_repository.h b/training/kbest_repository.h
new file mode 100644
index 00000000..0345394a
--- /dev/null
+++ b/training/kbest_repository.h
@@ -0,0 +1,19 @@
+#ifndef _KBEST_REPOSITORY_H_
+#define _KBEST_REPOSITORY_H_
+
+#include <vector>
+#include "wordid.h"
+#include "ns.h"
+#include "sparse_vector.h"
+
+class KBestRepository {
+  struct HypInfo {
+    std::vector<WordID> words;
+    SparseVector<double> x;
+    SufficientStats score_stats;
+  };
+
+  std::vector<HypInfo> candidates;
+};
+
+#endif