24 lines
626 B
Python
24 lines
626 B
Python
import pickle
|
||
from config import logger
|
||
from config import test_user_id
|
||
from config import activity_analysis_path
|
||
|
||
logger.info('加载用户活跃度字典')
|
||
user_activity_dict = pickle.load(open(activity_analysis_path + 'results/user_activity_dict.pkl', 'rb'))
|
||
|
||
|
||
def user_activity_predict(user_id):
|
||
"""
|
||
用户活跃度预测
|
||
"""
|
||
if user_id not in user_activity_dict:
|
||
result = -1
|
||
else:
|
||
result = user_activity_dict[user_id]
|
||
|
||
return result
|
||
|
||
|
||
if __name__ == '__main__':
|
||
result = user_activity_predict(user_id=test_user_id)
|
||
print('用户ID：', test_user_id, '贡献度：', result) |