推薦系統:NDCG評價指標及其Python實現方法

除了推薦任務,在二分類任務中,也能夠計算該指標,代碼以下:python # ndcg def get_dcg(y_pred, y_true, k): #注意y_pred與y_true必須是一一對應的,而且y_pred越大越接近label=1(用相關性的說法就是,與label=1越相關) df = pd.DataFrame({"y_pred":y_pred, "y_true":y_t
相關文章
相關標籤/搜索