реализация функции enrich_review
This commit is contained in:
parent
5bc569a75e
commit
bb19d92400
12
main.py
12
main.py
@ -40,7 +40,7 @@ def validate_rating(rating: int):
|
|||||||
return res
|
return res
|
||||||
|
|
||||||
|
|
||||||
def is_spam(text: str, spam_indicators: list):
|
def validate_rating(text: str, spam_indicators: list):
|
||||||
res = False
|
res = False
|
||||||
caps_count = 0
|
caps_count = 0
|
||||||
for i in spam_indicators:
|
for i in spam_indicators:
|
||||||
@ -58,7 +58,15 @@ def is_spam(text: str, spam_indicators: list):
|
|||||||
|
|
||||||
|
|
||||||
def enrich_review(review: dict, keywords_dict: dict, positive_words: set, negative_words: set):
|
def enrich_review(review: dict, keywords_dict: dict, positive_words: set, negative_words: set):
|
||||||
return dict
|
upgrade_review = {
|
||||||
|
"id": review.get("id"),
|
||||||
|
"rating": review.get("rating"),
|
||||||
|
"text": review.get("text"),
|
||||||
|
"clean_txt": clean_text(review.get("text")),
|
||||||
|
"sentiment": calculate_sentiment_score(review.get("text"), positive_words, negative_words),
|
||||||
|
"topics": extract_keywords(review.get("text"), keywords_dict)
|
||||||
|
}
|
||||||
|
return upgrade_review
|
||||||
|
|
||||||
|
|
||||||
def filter_valid_reviews(reviews: list):
|
def filter_valid_reviews(reviews: list):
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user