Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import numpy as np
- def recommend_hotel(A, userid):
- '''
- 向用户id为userid的用户推荐3家酒店
- :param A: 已经更新好了的矩阵A
- :param userid: 待推荐的userid,类型为整数
- :return: recommend
- '''
- #********* Begin *********#
- # 对矩阵A中userid对应的行进行升序排序
- ranklist =np.argsort(A[userid])
- #********* End *********#
- recommend = ranklist[-1:-4:-1]
- return recommend[-1], recommend[-2], recommend[-3]
Add Comment
Please, Sign In to add comment