реализация функции extract_keywords

This commit is contained in:
Максим Гетман 2026-04-22 22:19:05 +03:00
parent d88b07f613
commit bc4fd28a1d

15
main.py
View File

@ -15,7 +15,12 @@ def clean_text(text: str):
def extract_keywords(text: str, keywords_dict: dict): 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): def calculate_sentiment_score(text: str, positive_words: set, negative_words: set):
@ -93,10 +98,10 @@ def main():
"золотое", "слишком дорого", "цена не соответствует качеству", "дороже чем у других", "золотое", "слишком дорого", "цена не соответствует качеству", "дороже чем у других",
"наценка", "обдираловка", "цена выросла", "денег не стоит" "наценка", "обдираловка", "цена выросла", "денег не стоит"
] ]
keywords_dict: { keywords_dict = {
"quality": [positive_words + negative_words], "quality": list(positive_words) + list(negative_words),
"delivery": [delivery_words], "delivery": delivery_words,
"price": [price_words] "price": price_words
} }
print("\n======== Введите название файла с форматом .csv или используйте по умолчанию (Enter) ========") print("\n======== Введите название файла с форматом .csv или используйте по умолчанию (Enter) ========")
name_report = input() name_report = input()