реализация функции extract_keywords
This commit is contained in:
parent
d88b07f613
commit
bc4fd28a1d
15
main.py
15
main.py
@ -15,7 +15,12 @@ def clean_text(text: str):
|
||||
|
||||
|
||||
def extract_keywords(text: str, keywords_dict: dict):
|
||||
return list
|
||||
empty_list = []
|
||||
for k, v in keywords_dict.items():
|
||||
for i in v:
|
||||
if i in text:
|
||||
empty_list.append(k)
|
||||
return (list(set(empty_list)))
|
||||
|
||||
|
||||
def calculate_sentiment_score(text: str, positive_words: set, negative_words: set):
|
||||
@ -93,10 +98,10 @@ def main():
|
||||
"золотое", "слишком дорого", "цена не соответствует качеству", "дороже чем у других",
|
||||
"наценка", "обдираловка", "цена выросла", "денег не стоит"
|
||||
]
|
||||
keywords_dict: {
|
||||
"quality": [positive_words + negative_words],
|
||||
"delivery": [delivery_words],
|
||||
"price": [price_words]
|
||||
keywords_dict = {
|
||||
"quality": list(positive_words) + list(negative_words),
|
||||
"delivery": delivery_words,
|
||||
"price": price_words
|
||||
}
|
||||
print("\n======== Введите название файла с форматом .csv или используйте по умолчанию (Enter) ========")
|
||||
name_report = input()
|
||||
|
||||
Loading…
Reference in New Issue
Block a user