explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-11-10 19:57:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

* Fix bounds checking on entities

Browse Source
This commit is contained in:
Matthew Honnibal 2015-06-23 04:35:08 +02:00
parent 43ef5ddea5
commit ee3e56f27b
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
spacy/syntax/stateclass.pyx
View File

@ -35,9 +35,9 @@ cdef class StateClass:
cdef int E(self, int i) nogil:
if self._e_i <= 0 or self._e_i >= self.length:
return -1
if i <= 0 or i >= self.length:
return -1
return 0
if i < 0 or i >= self.length:
return 0
return self._ents[self._e_i-1].start
cdef int L(self, int i, int idx) nogil:
@ -203,11 +203,10 @@ cdef class StateClass:
self._sent[head].l_kids -= 1
cdef void open_ent(self, int label) nogil:
if 0 <= self._e_i < self.length:
self._ents[self._e_i].start = self.B(0)
self._ents[self._e_i].label = label
self._ents[self._e_i].end = -1
self._e_i += 1
self._ents[self._e_i].start = self.B(0)
self._ents[self._e_i].label = label
self._ents[self._e_i].end = -1
self._e_i += 1
cdef void close_ent(self) nogil:
self._ents[self._e_i-1].end = self.B(0)+1