119 lines
4.2 KiB
Python
119 lines
4.2 KiB
Python
|
import io
|
|||
|
|
import logging
|
|||
|
|
import sys
|
|||
|
|
import time
|
|||
|
|
from typing import List
|
|||
|
|
|
|||
|
|
import numpy as np
|
|||
|
|
from apscheduler.schedulers.background import BackgroundScheduler
|
|||
|
|
from apscheduler.triggers.cron import CronTrigger
|
|||
|
|
from fastapi import HTTPException, APIRouter
|
|||
|
|
|
|||
|
|
from app.service.recommend.service import load_resources, matrix_data
|
|||
|
|
|
|||
|
|
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
|
|||
|
|
logger = logging.getLogger()
|
|||
|
|
router = APIRouter()
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.on_event("startup")
|
|||
|
|
async def startup_event():
|
|||
|
|
# 初始加载
|
|||
|
|
load_resources()
|
|||
|
|
|
|||
|
|
# 配置定时任务
|
|||
|
|
scheduler = BackgroundScheduler()
|
|||
|
|
scheduler.add_job(
|
|||
|
|
load_resources,
|
|||
|
|
trigger=CronTrigger(hour=0, minute=30),
|
|||
|
|
name="每日资源刷新"
|
|||
|
|
)
|
|||
|
|
scheduler.start()
|
|||
|
|
logger.info("定时任务已启动")
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.get("/recommend/{user_id}/{category}/{num_recommendations}", response_model=List[str])
|
|||
|
|
async def get_recommendations(user_id: int, category: str, num_recommendations: int = 10):
|
|||
|
|
"""
|
|||
|
|
:param user_id: 4
|
|||
|
|
:param category: female_skirt
|
|||
|
|
:param num_recommendations: 1
|
|||
|
|
:return:
|
|||
|
|
[
|
|||
|
|
"aida-sys-image/images/female/skirt/903000017.jpg"
|
|||
|
|
]
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
start_time = time.time()
|
|||
|
|
cache_key = (user_id, category)
|
|||
|
|
|
|||
|
|
# 检查缓存
|
|||
|
|
if cache_key in matrix_data["cached_scores"]:
|
|||
|
|
processed_inter, processed_feat = matrix_data["cached_scores"][cache_key]
|
|||
|
|
valid_sketch_idxs_inter = matrix_data["cached_valid_idxs"][cache_key]
|
|||
|
|
else:
|
|||
|
|
# 实时计算逻辑（同原代码）
|
|||
|
|
user_idx_inter = matrix_data["user_index_interaction"].get(user_id)
|
|||
|
|
user_idx_feature = matrix_data["user_index_feature"].get(user_id)
|
|||
|
|
|
|||
|
|
category_iids = matrix_data["category_to_iids"].get(category, [])
|
|||
|
|
valid_sketch_idxs_inter = [
|
|||
|
|
idx for iid, idx in matrix_data["sketch_index_interaction"].items()
|
|||
|
|
if iid in category_iids
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
# 处理交互分数
|
|||
|
|
raw_inter_scores = []
|
|||
|
|
if user_idx_inter is not None and valid_sketch_idxs_inter:
|
|||
|
|
raw_inter_scores = matrix_data["interaction_matrix"][user_idx_inter, valid_sketch_idxs_inter]
|
|||
|
|
processed_inter = raw_inter_scores * 0.7
|
|||
|
|
|
|||
|
|
# 处理特征分数
|
|||
|
|
valid_sketch_idxs_feature = [
|
|||
|
|
idx for iid, idx in matrix_data["sketch_index_feature"].items()
|
|||
|
|
if iid in category_iids
|
|||
|
|
]
|
|||
|
|
raw_feat_scores = []
|
|||
|
|
if user_idx_feature is not None and valid_sketch_idxs_feature:
|
|||
|
|
raw_feat_scores = matrix_data["feature_matrix"][user_idx_feature, valid_sketch_idxs_feature]
|
|||
|
|
raw_feat_scores = (raw_feat_scores - np.min(raw_feat_scores)) / (
|
|||
|
|
np.max(raw_feat_scores) - np.min(raw_feat_scores) + 1e-8)
|
|||
|
|
processed_feat = raw_feat_scores * 0.3
|
|||
|
|
else:
|
|||
|
|
processed_feat = np.array([])
|
|||
|
|
|
|||
|
|
# 更新缓存
|
|||
|
|
matrix_data["cached_scores"][cache_key] = (processed_inter, processed_feat)
|
|||
|
|
matrix_data["cached_valid_idxs"][cache_key] = valid_sketch_idxs_inter
|
|||
|
|
|
|||
|
|
# 合并分数
|
|||
|
|
final_scores = processed_inter + processed_feat
|
|||
|
|
valid_sketch_idxs = matrix_data["cached_valid_idxs"][cache_key]
|
|||
|
|
|
|||
|
|
# 概率采样
|
|||
|
|
scores = np.array(final_scores)
|
|||
|
|
|
|||
|
|
# 调整后的概率转换（带温度控制的softmax）
|
|||
|
|
def calibrated_softmax(scores, temperature=1.0):
|
|||
|
|
scores = scores / temperature
|
|||
|
|
scale = scores - max(scores)
|
|||
|
|
exps = np.exp(scale)
|
|||
|
|
return exps / np.sum(exps)
|
|||
|
|
|
|||
|
|
probs = calibrated_softmax(scores, 0.07)
|
|||
|
|
|
|||
|
|
chosen_indices = np.random.choice(
|
|||
|
|
len(valid_sketch_idxs),
|
|||
|
|
size=min(num_recommendations, len(valid_sketch_idxs)),
|
|||
|
|
p=probs,
|
|||
|
|
replace=False
|
|||
|
|
)
|
|||
|
|
recommendations = [matrix_data["iid_to_sketch"][valid_sketch_idxs[idx]] for idx in chosen_indices]
|
|||
|
|
|
|||
|
|
logger.info(f"推荐生成完成，耗时: {time.time() - start_time:.2f}秒")
|
|||
|
|
return recommendations
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"推荐失败: {str(e)}", exc_info=True)
|
|||
|
|
raise HTTPException(status_code=500, detail=str(e))
|