初始矩阵包含链接预测系统中1
的已知链接(0
未知和错过的条目)。矩阵分解的输出是要计算的错过条目的预测值AUC
:
- 隐藏
20%
已知链接(矩阵中带有to20%
的条目集)1
0
- 对 Factorization 的输出进行排序并丢弃用于 train (
80%
of1
)的索引 - 设置
N
为隐藏值的数量 - 获取
N
最高预测值并检查它们是否是隐藏值(类标签设置为1
)或不是(类标签设置为0
) AUC
使用N
最高预测进行计算
我知道perfcurve
在 matlab 计算中AUC
,但我需要确定上述过程以提供标记数据perfcurve
。
任何评论都非常感谢。