mirror of https://github.com/explosion/spaCy.git
* Reenable averaging
This commit is contained in:
parent
df8179ca4f
commit
5cd3ed42d4
|
@ -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):
|
||||||
|
|
Loading…
Reference in New Issue