我正在使用简单的用户项矩阵在LightFM中开发纯协作过滤模型。
当我的训练矩阵的行数多于测试矩阵时,我遇到此错误:“user_features中的功能数量不正确”
这似乎是由lightfm.py中的第789行引起的:
if not user_features.shape[1] == self.user_embeddings.shape[0]:
raise ValueError('Incorrect number of features in user_features')
调查此事 - 看起来它是由lightfm.py中的第778行引起的:
n_users, n_items = test_interactions.shape
这表明测试数据集的大小应该与列车匹配。并使它们大小相同可以解决我的错误。
我如何说服LightFM让我训练和测试不同大小的数据集?