vanessbut commited on
Commit
8e49c01
·
1 Parent(s): 5a38489

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

Browse files
Files changed (1) hide show
  1. utils/utils.py +1 -1
utils/utils.py CHANGED
@@ -58,7 +58,7 @@ def get_candidates(text, nlp, min_df=0.0, ngram_range=(1, 3), max_words=None):
58
  if token.pos_ == "NOUN" and (token.text == token.lemma_ or not (token.text in noun_lemmas)):
59
  nouns.add(token.text)
60
  #print(nouns)
61
- nouns = nouns.union(noun_lemmas)
62
 
63
  # Объединение.
64
  with_nouns = nouns.union(noun_phrases)
 
58
  if token.pos_ == "NOUN" and (token.text == token.lemma_ or not (token.text in noun_lemmas)):
59
  nouns.add(token.text)
60
  #print(nouns)
61
+ nouns = noun_lemmas #nouns.union(noun_lemmas)
62
 
63
  # Объединение.
64
  with_nouns = nouns.union(noun_phrases)