公开(公告)号：CN113112148A

公开(公告)日：2021-07-13

申请号：CN202110384899.1

申请日：2021-04-09

Applicant: 北京邮电大学

Inventor： 李凯元 ,  冯志超 ,  陈昱硕 ,  侯宇蓬 ,  王鹏飞 ,  赵鑫

IPC: G06Q10/06 ,  G06Q30/06 ,  G06F16/951 ,  G06F16/9535

Abstract: 本公开提供一种推荐系统模型评测结果的评测方法及电子设备。应用于PyTorch推荐系统框架，通过获取单个批次内每个用户对候选商品的一维分数张量，根据评测方式进行对应重排处理，得到正样本张量均位于负样本张量之前的二维分数张量；生成二维分数张量的列索引的乱序排列，得到重排二维分数张量；并行计算单个批次内每个用户的top‑k商品的原始索引张量；循环前述步骤得到所有用户的top‑k商品的原始索引；计算所有正样本张量与所有用户的top‑k商品的交集在所有用户的top‑k商品的原始索引中的布尔索引张量；对布尔索引张量和所有正样本张量进行计算，得到所有相关top‑k指标的值。测评准确，且兼顾运行速度和空间。

公开(公告)号：CN113112148B

公开(公告)日：2022-08-05

申请号：CN202110384899.1

申请日：2021-04-09

Applicant: 北京邮电大学

Inventor： 李凯元 ,  冯志超 ,  陈昱硕 ,  侯宇蓬 ,  王鹏飞 ,  赵鑫

IPC: G06Q10/06 ,  G06Q30/06 ,  G06F16/951 ,  G06F16/9535

Abstract: 本公开提供一种推荐系统模型评测结果的评测方法及电子设备。应用于PyTorch推荐系统框架，通过获取单个批次内每个用户对候选商品的一维分数张量，根据评测方式进行对应重排处理，得到正样本张量均位于负样本张量之前的二维分数张量；生成二维分数张量的列索引的乱序排列，得到重排二维分数张量；并行计算单个批次内每个用户的top‑k商品的原始索引张量；循环前述步骤得到所有用户的top‑k商品的原始索引；计算所有正样本张量与所有用户的top‑k商品的交集在所有用户的top‑k商品的原始索引中的布尔索引张量；对布尔索引张量和所有正样本张量进行计算，得到所有相关top‑k指标的值。测评准确，且兼顾运行速度和空间。