Mean Percentage Ranking (MPR) 實現

 Mean Percentage Ranking (MPR) 是推薦系統的一個常用的評估函數,在IMF和LMF論文中都使用這一評價指標。

                                                                       

mpr實現:

import numpy as np


def MPR(R_hat,R):
    '''

    :param R_hat: user dot item
    :param R: ratting
    :return: mean percentage rank
    '''
    rank = 0.0
    R_sum=np.sum(R)
    n_games=R.shape[1]
    R_hat_rank=np.argsort(np.argsort(-R_hat,axis=1))
    A=R*(R_hat_rank/n_games)
    rank = np.sum(A) / R_sum
    return rank

其中,使用2次np.argsort是爲了得到元素在數組中的順序名次。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章