Исправление кандидатов.

utils/utils.py

@@ -51,12 +51,16 @@ def get_candidates(text, nlp, min_df=0.0, ngram_range=(1, 3), max_words=None):
     for token in nlp_result:
         if token.pos_ == "NOUN":
             noun_lemmas.add(token.lemma_) # Для одного слова всё-таки бессмысленно хранить форму.
-    #print(noun_lemmas)
-            
-    nouns = set()
-    for token in nlp_result:
-        if token.pos_ == "NOUN" and (token.text == token.lemma_ or not (token.text in noun_lemmas)):
-            nouns.add(token.text)
+    print(noun_lemmas)
+    
+    #nouns = set()
+    #noun_lemmas = set()
+
+    # Сначала составные слова.
+    #for token in nlp_result:
+    #    if token.pos_ == "NOUN":
+    #        noun_lemmas.add(token.lemma_) # Для одного слова всё-таки бессмысленно хранить форму.
+    #        nouns.add(token.text)
     #print(nouns)
     nouns = noun_lemmas #nouns.union(noun_lemmas)