Merge branch 'master' of github.com:pks/cdec-dtrain

1 files changed, 19 insertions, 3 deletions

diff --git a/python/src/_cdec.pyx b/python/src/_cdec.pyx
index 698a66f6..cccfec0b 100644
--- a/python/src/_cdec.pyx
+++ b/python/src/_cdec.pyx
@@ -6,15 +6,15 @@ cimport decoder
 include "vectors.pxi"
 include "hypergraph.pxi"
 include "lattice.pxi"
+include "mteval.pxi"
 
 SetSilent(True)
 
-class ParseFailed(Exception):
-    pass
+class ParseFailed(Exception): pass
 
 cdef class Decoder:
     cdef decoder.Decoder* dec
-    cdef public DenseVector weights
+    cdef DenseVector weights
 
     def __cinit__(self, char* config):
         decoder.register_feature_functions()
@@ -27,6 +27,22 @@ cdef class Decoder:
     def __dealloc__(self):
         del self.dec
 
+    property weights:
+        def __get__(self):
+            return self.weights
+
+        def __set__(self, weights):
+            if isinstance(weights, DenseVector):
+                self.weights.vector[0] = (<DenseVector> weights).vector[0]
+            elif isinstance(weights, SparseVector):
+                self.weights.vector.clear()
+                ((<SparseVector> weights).vector[0]).init_vector(self.weights.vector)
+            elif isinstance(weights, dict):
+                for fname, fval in weights.items():
+                    self.weights[fname] = fval
+            else:
+                raise TypeError('cannot initialize weights with %s' % type(weights))
+
     def read_weights(self, cfg):
         with open(cfg) as fp:
             for line in fp: