rewind

read_extract.py

@@ -224,9 +224,9 @@ def get_keywords(datatype, task, field, pos_text, neg_text):
 	print('Extracting keywords...')
 	datatype, task, field = datatype.lower(), task.lower(), field.lower()
 
-	#stop_words = set(stopwords.words('english'))
-	stops = nltk.corpus.stopwords.words('english')
-	stop_words = set(stops)
+	stop_words = set(stopwords.words('english'))
+	#stops = nltk.corpus.stopwords.words('english')
+	#stop_words = set(stops)
 	stop_words = stop_words.union({'license', 'licensing', 'licensor', 'copyright', 'copyrights', 'patent'})
 
 	pos_kw_model = KeyBERT()