我在进行集合预测时遇到了困难。最初我想使用共现方法来解决这个问题,但尝试之后我开始怀疑这是否是合适的工具。
我有一个数据库,里面有一些数据(每列对应一个特定的项目,每行对应一个集合),例如:
data:[[1 0 1 1] [1 1 1 1] [1 0 1 1]]
我计算了共现矩阵:
cooccur_matrix:[[0 1 3 3] [1 0 1 1] [3 1 0 3] [3 1 3 0]]
现在我有一个不完整的集合:
target:[1 0 1 1]
我的共现矩阵与这个集合的点积是:
prediction:[6 3 6 6]
但这完全不是我想要的结果。我希望得到的是类似这样的结果:
prediction:[1 0.33 1 1]
或者是:
prediction:[0 0.33 0 0]
对于我做错的地方有什么建议吗?我对机器学习的概念还比较新手,这看起来像是一个相当简单的问题。
回答:
看起来我真正需要的是将余弦相似度应用于共现矩阵。这似乎效果很好(但解决起来非常困难)。