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

* Add from_array function to Doc.

Browse Source
This commit is contained in:
Matthew Honnibal 2015-07-16 17:46:11 +02:00
parent 2a5d050134
commit a6f401580d
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
spacy/tokens/doc.pyx
View File

@ -370,3 +370,23 @@ cdef class Doc:
# Return the merged Python object # Return the merged Python object
return self[start] return self[start]
def from_array(self, attrs, array):
cdef int i
cdef attr_id_t attr_id
cdef TokenC* tokens = self.data
for attr_id in attrs:
if attr_id == HEAD:
for i in range(length):
tokens[i].head = values[i]
elif attr_id == TAG:
for i in range(length):
tokens[i].tag = values[i]
elif attr_id == DEP:
for i in range(length):
tokens[i].dep = values[i]
elif attr_id == ENT_IOB:
for i in range(length):
tokens[i].ent_iob = values[i]
elif attr_id == ENT_TYPE:
for i in range(length):
tokens[i].ent_type = values[i]