* Reenable averaging

This commit is contained in:
Matthew Honnibal 2015-06-29 16:44:42 +02:00
parent df8179ca4f
commit 5cd3ed42d4
1 changed files with 1 additions and 1 deletions

View File

@ -29,7 +29,7 @@ cdef class TheanoModel(Model):
self.model_loc = model_loc self.model_loc = model_loc
def predict(self, Example eg): def predict(self, Example eg):
self.input_layer.fill(eg.embeddings, eg.atoms, use_avg=False) self.input_layer.fill(eg.embeddings, eg.atoms, use_avg=True)
theano_scores = self.predict_func(eg.embeddings)[0] theano_scores = self.predict_func(eg.embeddings)[0]
cdef int i cdef int i
for i in range(self.n_classes): for i in range(self.n_classes):