summaryrefslogtreecommitdiff
path: root/decoder/apply_models.h
diff options
context:
space:
mode:
Diffstat (limited to 'decoder/apply_models.h')
-rw-r--r--decoder/apply_models.h20
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..d6d8b34a
--- /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 IntersectionConfiguration {
+ const int algorithm; // 0 = full intersection, 1 = cube pruning
+ const int pop_limit; // max number of pops off the heap at each node
+ IntersectionConfiguration(int alg, int k) : algorithm(alg), pop_limit(k) {}
+};
+
+void ApplyModelSet(const Hypergraph& in,
+ const SentenceMetadata& smeta,
+ const ModelSet& models,
+ const IntersectionConfiguration& config,
+ Hypergraph* out);
+
+#endif