lanyuelvyun的博客

原创 推荐系统：NDCG评价指标及其Python实现方法

除了推荐任务，在二分类任务中，也可以计算该指标，代码如下：# ndcgdef 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_true})...