summaryrefslogtreecommitdiff
path: root/python/cdec/sa/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/cdec/sa/__init__.py')
-rw-r--r--python/cdec/sa/__init__.py25
1 files changed, 25 insertions, 0 deletions
diff --git a/python/cdec/sa/__init__.py b/python/cdec/sa/__init__.py
new file mode 100644
index 00000000..14ba5ecb
--- /dev/null
+++ b/python/cdec/sa/__init__.py
@@ -0,0 +1,25 @@
+from cdec.sa._sa import make_lattice, decode_lattice, decode_sentence,\
+ encode_words, decode_words, isvar,\
+ SuffixArray, DataArray, LCP, Precomputation, Alignment, BiLex,\
+ HieroCachingRuleFactory, Sampler, Scorer
+from cdec.sa.extractor import GrammarExtractor
+
+_SA_FEATURES = []
+_SA_ANNOTATORS = {}
+_SA_CONFIGURE = []
+
+def feature(fn):
+ _SA_FEATURES.append(fn)
+ return fn
+
+def annotator(fn):
+ _SA_ANNOTATORS[fn.__name__] = fn
+
+def annotate(sentence):
+ meta = {}
+ for name, fn in _SA_ANNOTATORS.iteritems():
+ meta[name] = fn(sentence)
+ return meta
+
+def configure(fn):
+ _SA_CONFIGURE.append(fn)