реализация функции 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):
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()