Fix numpy.zeros() dtype for Doc.from_array

This commit is contained in:
Adriane Boyd 2020-06-16 20:35:45 +02:00
parent 9aff317ca7
commit a6abdfbc3c
1 changed files with 1 additions and 1 deletions

View File

@ -471,7 +471,7 @@ doc = nlp.make_doc("London is a big city in the United Kingdom.")
print("Before", doc.ents) # [] print("Before", doc.ents) # []
header = [ENT_IOB, ENT_TYPE] header = [ENT_IOB, ENT_TYPE]
attr_array = numpy.zeros((len(doc), len(header))) attr_array = numpy.zeros((len(doc), len(header)), dtype="uint64")
attr_array[0, 0] = 3 # B attr_array[0, 0] = 3 # B
attr_array[0, 1] = doc.vocab.strings["GPE"] attr_array[0, 1] = doc.vocab.strings["GPE"]
doc.from_array(header, attr_array) doc.from_array(header, attr_array)