harshithakr commited on
Commit
232f66c
·
1 Parent(s): 31ff26f

Update kbert_topics.py

Browse files
Files changed (1) hide show
  1. kbert_topics.py +1 -1
kbert_topics.py CHANGED
@@ -19,7 +19,7 @@ with open("common_words_v2.txt", "r") as file:
19
 
20
  def extract_topics(event_value):
21
  keywords = kw_model.extract_keywords(bert_preprocess(event_value), top_n = 10)
22
- keywords_2 = kw_model.extract_keywords(bert_preprocess(event_value), top_n = 5, keyphrase_ngram_range=(2))
23
 
24
  entity_remove = {'CARDINAL', 'DATE', 'GPE', 'LANGUAGE', 'LOC',
25
  'NORP', 'ORDINAL', 'PERCENT', 'PERSON', 'QUANTITY', 'TIME'}
 
19
 
20
  def extract_topics(event_value):
21
  keywords = kw_model.extract_keywords(bert_preprocess(event_value), top_n = 10)
22
+ keywords_2 = kw_model.extract_keywords(bert_preprocess(event_value), top_n = 5, keyphrase_ngram_range=(1,2))
23
 
24
  entity_remove = {'CARDINAL', 'DATE', 'GPE', 'LANGUAGE', 'LOC',
25
  'NORP', 'ORDINAL', 'PERCENT', 'PERSON', 'QUANTITY', 'TIME'}