分组示例的正确学习算法

数据挖掘 机器学习 分类 算法
2022-02-26 07:05:55

我需要了解(最有可能)从一组中挑选哪个项目。我拥有的数据由 10 个项目组成,其中一个被选中(或者如果两个都没有被选中)。

我目前的方法是忽略这些组,并使用平面列表将所有项目分类为已挑选或未挑选并使用一些分类算法,但我想知道是否有办法使用从该特定项目中挑选项目的知识团体。

1个回答

M是每个组的大小。你想学习函数f这需要M矢量图vi,i[1,M] 并产生一个向量v^可以是 0 或其中之一vi. 你可以训练一个 DNN 来学习f通过按如下方式组织您的训练数据:

Input              Label
[batchSize, M, 1]  [batchSize,1]

优化预测向量和标签向量之间减少的 RMS 误差。然后,您的 DNN 将学习组信息。