LightFM训练集需要与测试集大小相同吗?

时间:2018-03-21 12:28:59

标签: python machine-learning recommendation-engine collaborative-filtering

我正在使用简单的用户项矩阵在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让我训练和测试不同大小的数据集?

0 个答案:

没有答案