diff --git a/app/services/daily_weight_resizer.py b/app/services/daily_weight_resizer.py index 81aeabb..c7bf946 100644 --- a/app/services/daily_weight_resizer.py +++ b/app/services/daily_weight_resizer.py @@ -5,7 +5,6 @@ import numpy as np from collections import defaultdict import app -from app.models import db_w2v_mapper from app.repositories.managed_action_log_repository import ManagedActionLogRepository from app.services import weight_strategy from app.enum.action_type import ActionType @@ -64,15 +63,13 @@ async def resize_weight( directors = meta.get('director', {}) countries = meta.get('country', {}) - for _, genre_name in genres.items(): - translated = db_w2v_mapper.translate_genre(genre_name) - if translated: - genre_dict[translated] += resized_weight - for _, actor_name in actors.items(): + for _, genre_name in genres: + genre_dict[genre_name] += resized_weight + for _, actor_name in actors: actor_dict[actor_name] += resized_weight - for _, director_name in directors.items(): + for _, director_name in directors: director_dict[director_name] += resized_weight - for _, country_name in countries.items(): + for _, country_name in countries: country_dict[country_name] += resized_weight diff --git a/app/util/weight_aging.py b/app/util/weight_aging.py index c9997ee..592de8c 100644 --- a/app/util/weight_aging.py +++ b/app/util/weight_aging.py @@ -4,7 +4,7 @@ def exponential_decay_weight(original_weight: float, event_timestamp_ms: int, lambda_: float = 0.1) -> float: current_timestamp_ms = int(time.time() * 1000) delta_ms = current_timestamp_ms - event_timestamp_ms - delta_days = delta_ms / (1000 * 60 * 60 * 24) + delta_days = int(delta_ms / (1000 * 60 * 60 * 24)) decayed_weight = original_weight * math.exp(-lambda_ * delta_days) - return round(decayed_weight, 2) + return decayed_weight