Update word_frequency_functions.py

cclauss · web-flow · commit 5f814507bfd6 · 2020-06-25T12:35:57.000+02:00
diff --git a/machine_learning/word_frequency_functions.py b/machine_learning/word_frequency_functions.py
@@ -81,7 +81,7 @@ def document_frequency(term: str, corpus: str) -> int:
     documents = corpus_without_punctuation.split("\n")
     lowercase_documents = [document.lower() for document in documents]
     return (
-        len([document for document in lowercase_documents if term.lower() in document]),
+        len([doc for doc in lowercase_documents if term.lower() in doc]),
         len(documents),
     )
 

Original file line number	Diff line number	Diff line change
`@@ -81,7 +81,7 @@ def document_frequency(term: str, corpus: str) -> int:`
`81`	`81`	`documents = corpus_without_punctuation.split("\n")`
`82`	`82`	`lowercase_documents = [document.lower() for document in documents]`
`83`	`83`	`return (`
`84`		`- len([document for document in lowercase_documents if term.lower() in document]),`
	`84`	`+ len([doc for doc in lowercase_documents if term.lower() in doc]),`
`85`	`85`	`len(documents),`
`86`	`86`	`)`
`87`	`87`