реализация функции 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):
|
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()
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user