Spaces:
Runtime error
Runtime error
Commit ·
232f66c
1
Parent(s): 31ff26f
Update kbert_topics.py
Browse files- 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'}
|