FancyKing

第2关:酒店智能推荐

Apr 28th, 2020
242
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.49 KB | None | 0 0
  1. import numpy as np
  2.  
  3. def recommend_hotel(A, userid):
  4.     '''
  5.    向用户id为userid的用户推荐3家酒店
  6.    :param A: 已经更新好了的矩阵A
  7.    :param userid: 待推荐的userid,类型为整数
  8.    :return: recommend
  9.    '''
  10.     #********* Begin *********#
  11.     # 对矩阵A中userid对应的行进行升序排序
  12.     ranklist =np.argsort(A[userid])
  13.  
  14.     #********* End *********#
  15.     recommend = ranklist[-1:-4:-1]
  16.     return recommend[-1], recommend[-2], recommend[-3]
Add Comment
Please, Sign In to add comment