Ezi Ozoani commited on
Commit
13af81f
·
1 Parent(s): 967976f

maybe this fixes it

Browse files
Files changed (1) hide show
  1. read_extract.py +2 -1
read_extract.py CHANGED
@@ -224,7 +224,8 @@ def get_keywords(datatype, task, field, pos_text, neg_text):
224
  print('Extracting keywords...')
225
  datatype, task, field = datatype.lower(), task.lower(), field.lower()
226
 
227
- stop_words = set(stopwords.words('english'))
 
228
  stop_words = stop_words.union({'license', 'licensing', 'licensor', 'copyright', 'copyrights', 'patent'})
229
 
230
  pos_kw_model = KeyBERT()
 
224
  print('Extracting keywords...')
225
  datatype, task, field = datatype.lower(), task.lower(), field.lower()
226
 
227
+ #stop_words = set(stopwords.words('english'))
228
+ stop_words = nltk.corpus.stopwords.words('english')
229
  stop_words = stop_words.union({'license', 'licensing', 'licensor', 'copyright', 'copyrights', 'patent'})
230
 
231
  pos_kw_model = KeyBERT()