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

Find span sentence when only sentence boundaries (no parser)

Browse Source
This commit is contained in:
Thomas Opsomer 2018-02-14 22:18:54 +01:00
parent c2b0910db4
commit b902731313
1 changed files with 25 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
spacy/tokens/span.pyx
View File

@ -285,9 +285,13 @@ cdef class Span:
def __get__(self):
if 'sent' in self.doc.user_span_hooks:
return self.doc.user_span_hooks['sent'](self)
# This should raise if we're not parsed.
# This should raise if we're not parsed
# or doesen't have any sbd component :)
self.doc.sents
# if doc is parsed we can use the deps to find the sentence
# otherwise we use the `sent_start` token attribute
cdef int n = 0
if self.doc.is_parsed:
root = &self.doc.c[self.start]
while root.head != 0:
root += root.head
@ -295,6 +299,19 @@ cdef class Span:
if n >= self.doc.length:
raise RuntimeError
return self.doc[root.l_edge:root.r_edge + 1]
else:
# find start of the sentence
start = self.start
while not self.doc.c[start].sent_start and start > 0:
start += -1
# find end of the sentence
end = self.end
while not self.doc.c[end].sent_start:
end += 1
if n >= self.doc.length:
break
#
return self.doc[start:end]
property has_vector:
"""RETURNS (bool): Whether a word vector is associated with the object.