mirror of https://github.com/explosion/spaCy.git
146 lines
4.5 KiB
Cython
146 lines
4.5 KiB
Cython
from __future__ import unicode_literals
|
|
from collections import defaultdict
|
|
|
|
from ..structs cimport TokenC, LexemeC
|
|
from ..typedefs cimport flags_t, attr_t
|
|
from ..attrs cimport attr_id_t
|
|
from ..parts_of_speech cimport univ_pos_t
|
|
|
|
|
|
cdef class Span:
|
|
"""A slice from a Doc object."""
|
|
def __cinit__(self, Doc tokens, int start, int end, int label=0):
|
|
if start < 0:
|
|
start = tokens.length - start
|
|
if end < 0:
|
|
end = tokens.length - end
|
|
self._seq = tokens
|
|
self.start = start
|
|
self.end = end
|
|
self.label = label
|
|
|
|
def __richcmp__(self, Span other, int op):
|
|
# Eq
|
|
if op == 0:
|
|
return self.start < other.start
|
|
elif op == 1:
|
|
return self.start <= other.start
|
|
elif op == 2:
|
|
return self.start == other.start and self.end == other.end
|
|
elif op == 3:
|
|
return self.start != other.start or self.end != other.end
|
|
elif op == 4:
|
|
return self.start > other.start
|
|
elif op == 5:
|
|
return self.start >= other.start
|
|
|
|
def __len__(self):
|
|
if self.end < self.start:
|
|
return 0
|
|
return self.end - self.start
|
|
|
|
def __getitem__(self, int i):
|
|
if i < 0:
|
|
return self._seq[self.end + i]
|
|
else:
|
|
return self._seq[self.start + i]
|
|
|
|
def __iter__(self):
|
|
for i in range(self.start, self.end):
|
|
yield self._seq[i]
|
|
|
|
def merge(self, unicode tag, unicode lemma, unicode ent_type):
|
|
self._seq.merge(self[0].idx, self[-1].idx + len(self[-1]), tag, lemma, ent_type)
|
|
|
|
property text:
|
|
def __get__(self):
|
|
return u' '.join([t.text for t in self])
|
|
|
|
property text_with_ws:
|
|
def __get__(self):
|
|
return u''.join([t.text_with_ws for t in self])
|
|
|
|
property root:
|
|
"""The first ancestor of the first word of the span that has its head
|
|
outside the span.
|
|
|
|
For example:
|
|
|
|
>>> toks = nlp(u'I like New York in Autumn.')
|
|
|
|
Let's name the indices --- easier than writing "toks[4]" etc.
|
|
|
|
>>> i, like, new, york, in_, autumn, dot = range(len(toks))
|
|
|
|
The head of 'new' is 'York', and the head of 'York' is 'like'
|
|
|
|
>>> toks[new].head.orth_
|
|
'York'
|
|
>>> toks[york].head.orth_
|
|
'like'
|
|
|
|
Create a span for "New York". Its root is "York".
|
|
|
|
>>> new_york = toks[new:york+1]
|
|
>>> new_york.root.orth_
|
|
'York'
|
|
|
|
When there are multiple words with external dependencies, we take the first:
|
|
|
|
>>> toks[autumn].head.orth_, toks[dot].head.orth_
|
|
('in', like')
|
|
>>> autumn_dot = toks[autumn:]
|
|
>>> autumn_dot.root.orth_
|
|
'Autumn'
|
|
"""
|
|
def __get__(self):
|
|
# This should probably be called 'head', and the other one called
|
|
# 'gov'. But we went with 'head' elsehwhere, and now we're stuck =/
|
|
cdef const TokenC* start = &self._seq.data[self.start]
|
|
cdef const TokenC* end = &self._seq.data[self.end]
|
|
head = start
|
|
while start <= (head + head.head) < end and head.head != 0:
|
|
head += head.head
|
|
return self._seq[head - self._seq.data]
|
|
|
|
property lefts:
|
|
"""Tokens that are to the left of the Span, whose head is within the Span."""
|
|
def __get__(self):
|
|
for token in reversed(self): # Reverse, so we get the tokens in order
|
|
for left in token.lefts:
|
|
if left.i < self.start:
|
|
yield left
|
|
|
|
property rights:
|
|
"""Tokens that are to the right of the Span, whose head is within the Span."""
|
|
def __get__(self):
|
|
for token in self:
|
|
for right in token.rights:
|
|
if right.i >= self.end:
|
|
yield right
|
|
|
|
property subtree:
|
|
def __get__(self):
|
|
for word in self.lefts:
|
|
yield from word.subtree
|
|
yield from self
|
|
for word in self.rights:
|
|
yield from word.subtree
|
|
|
|
property orth_:
|
|
def __get__(self):
|
|
return ''.join([t.string for t in self]).strip()
|
|
|
|
property lemma_:
|
|
def __get__(self):
|
|
return ' '.join([t.lemma_ for t in self]).strip()
|
|
|
|
property string:
|
|
def __get__(self):
|
|
return ''.join([t.string for t in self])
|
|
|
|
property label_:
|
|
def __get__(self):
|
|
return self._seq.vocab.strings[self.label]
|
|
|