explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-10-31 07:57:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Allow DocBin to take list of Doc objects.

Browse Source
This commit is contained in:
Matthew Honnibal 2020-06-20 03:50:36 +02:00
parent 95df028758
commit 0d22c6e006
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/tokens/_serialize.py
View File

@ -39,7 +39,7 @@ class DocBin(object):
document from the DocBin. document from the DocBin.
""" """
def __init__(self, attrs=None, store_user_data=False): def __init__(self, attrs=None, store_user_data=False, docs=[]):
"""Create a DocBin object to hold serialized annotations. """Create a DocBin object to hold serialized annotations.
attrs (list): List of attributes to serialize. 'orth' and 'spacy' are attrs (list): List of attributes to serialize. 'orth' and 'spacy' are
@ -59,6 +59,8 @@ class DocBin(object):
self.user_data = [] self.user_data = []
self.strings = set() self.strings = set()
self.store_user_data = store_user_data self.store_user_data = store_user_data
for doc in docs:
self.add(docs)
def __len__(self): def __len__(self):
"""RETURNS: The number of Doc objects added to the DocBin.""" """RETURNS: The number of Doc objects added to the DocBin."""