explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-26 01:46:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

* Fix edge features

Browse Source
This commit is contained in:
Matthew Honnibal 2015-06-23 15:50:56 +02:00
parent 221e2e485f
commit 35c290bee4
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
spacy/syntax/stateclass.pyx
View File

@ -190,17 +190,27 @@ cdef class StateClass:
cdef int dist = head - child cdef int dist = head - child
self._sent[child].head = dist self._sent[child].head = dist
self._sent[child].dep = label self._sent[child].dep = label
cdef int i
if child > head: if child > head:
self._sent[head].r_kids += 1 self._sent[head].r_kids += 1
i = 0
while self.has_head(head) and i < self.length:
self._sent[head].r_edge = child
head = self.H(head)
i += 1 # Guard against infinite loops
else: else:
self._sent[head].l_kids += 1 self._sent[head].l_kids += 1
self._sent[head].l_edge = self._sent[child].l_edge
cdef void del_arc(self, int head, int child) nogil: cdef void del_arc(self, int h_i, int c_i) nogil:
cdef int dist = head - child cdef int dist = h_i - c_i
if child > head: cdef TokenC* h = &self._sent[h_i]
self._sent[head].r_kids -= 1 if c_i > h_i:
h.r_kids -= 1
h.r_edge = self.R_(h_i, h.r_kids-1).r_edge if h.r_kids >= 1 else h_i
else: else:
self._sent[head].l_kids -= 1 h.l_kids -= 1
h.l_edge = self.L_(h_i, h.l_kids-1).l_edge if h.l_kids >= 1 else h_i
cdef void open_ent(self, int label) nogil: cdef void open_ent(self, int label) nogil:
self._ents[self._e_i].start = self.B(0) self._ents[self._e_i].start = self.B(0)