diff options
Diffstat (limited to 'python/src/kbest.pxd')
| -rw-r--r-- | python/src/kbest.pxd | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/python/src/kbest.pxd b/python/src/kbest.pxd new file mode 100644 index 00000000..e339714a --- /dev/null +++ b/python/src/kbest.pxd @@ -0,0 +1,16 @@ +from libcpp.vector cimport vector +from utils cimport WordID +cimport hypergraph + +cdef extern from "decoder/viterbi.h": + cdef cppclass ESentenceTraversal: + pass + cdef cppclass ETreeTraversal: + pass + +cdef extern from "decoder/kbest.h" namespace "KBest": + cdef cppclass KBestDerivations[T, Traversal]: + cppclass Derivation: + T _yield "yield" + KBestDerivations(hypergraph.Hypergraph hg, unsigned k) + Derivation* LazyKthBest(unsigned v, unsigned k) |