Torch7 中 MultiLabelMarginCriterion 的替代方案

我需要解决一个多类别标签问题,即:每张测试图像可以被分配1到10个非互斥的标签。

然而,我在使用 MultiLabelMarginCriterion 时遇到了问题,因为它不被 cunn 支持。因此,我正在寻找替代方法。以下两种方法是否有效?

  1. 计算训练集中所有标签的排列组合(约150种),并训练一个分类器来识别这150个类别。然而,我认为测试集中新的标签组合(训练集中未出现的)将无法被识别。

  2. 使用 BCECriterion 训练10个独立的二元分类器。即:每个标签训练一个分类器。将每张测试图像通过每个分类器并合并结果。然而,训练大量的卷积神经网络非常耗时。


回答:

我找到的最佳解决方案是使用 MSECriterion,其中目标和预测都是由1和-1组成的数组,分别表示标签的存在或不存在。

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注