1 files changed, 20 insertions, 0 deletions
diff --git a/decoder/apply_models.h b/decoder/apply_models.h
new file mode 100644
index 00000000..08fce037
--- /dev/null
+++ b/decoder/apply_models.h
@@ -0,0 +1,20 @@
+#ifndef _APPLY_MODELS_H_
+#define _APPLY_MODELS_H_
+
+struct ModelSet;
+struct Hypergraph;
+struct SentenceMetadata;
+
+struct PruningConfiguration {
+  const int algorithm; // 0 = full intersection, 1 = cube pruning
+  const int pop_limit; // max number of pops off the heap at each node
+  explicit PruningConfiguration(int k) : algorithm(1), pop_limit(k) {}
+};
+
+void ApplyModelSet(const Hypergraph& in,
+                   const SentenceMetadata& smeta,
+                   const ModelSet& models,
+                   const PruningConfiguration& config,
+                   Hypergraph* out);
+
+#endif