Docstring improvements

This commit is contained in:
Samuel Sloniker 2021-08-17 12:29:39 -07:00
parent 275bb7f44b
commit 9eb602de65

View File

@ -6,7 +6,7 @@ import spacy
nlp = spacy.load('en_core_web_sm') nlp = spacy.load('en_core_web_sm')
def _listify(text): def _listify(text):
"""Convert a string to a list of lemmas. Internal use only.""" """Convert a string to a list of lemmas."""
return [string.lemma_.lower() for string in nlp(text) if string.lemma_[0] in 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'] return [string.lemma_.lower() for string in nlp(text) if string.lemma_[0] in 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ']
@ -61,7 +61,8 @@ class Classifier:
Parameters Parameters
---------- ----------
model : dict or list model : dict or list
A compiled or raw GPTC model. Please don't use raw models here. A compiled or raw GPTC model. Please don't use raw models except
during development.
Attributes Attributes
---------- ----------