diff options
Diffstat (limited to 'decoder/apply_models.h')
-rw-r--r-- | decoder/apply_models.h | 20 |
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 |