neural matrix completion:论文“可扩展神经矩阵完成”的源代码 tensorflow source code

神经矩阵完成 该存储库包含题为“可扩展神经矩阵完成”的论文的源代码(在ICASSP 2018上发布) 先决条件 - Tensorflow - Numpy 该代码已经在Ubuntu 18.04和MacOSX中进行了测试, - Python 3.5 - Tensorflow v.1.8 - Numpy v.1.14 您还应该能够与Tensorflow和Numpy的其他版本一起运行代码。 用法 准备数据 包含了MovieLens100K数据集的样本拆分(75%训练,5%验证,20%测试)(在文件夹./data/MovieLens100K/中): rating.npz:原始矩阵 train_mask.npz:训练蒙版,形状与原始矩阵相同 val_mask.npz:验证蒙版,形状与原始矩阵相同 test_mask.npz:测试遮罩,形状与原始矩阵相同 所有这些矩阵都保存为Numpy稀疏csr_