реализация функции 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
|
||||
|
||||
|
||||
def is_spam(text: str, spam_indicators: list):
|
||||
def validate_rating(text: str, spam_indicators: list):
|
||||
res = False
|
||||
caps_count = 0
|
||||
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):
|
||||
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):
|
||||
|
||||
Loading…
Reference in New Issue
Block a user