【发布时间】:2017-07-27 07:17:02
【问题描述】:
我有一组数据,每个条目有 9 个“特征”(正数范围从 1e-3 到 9e3),我需要使用每个条目的特征为条目选择 5 个可能的标签中的一个。
我知道我需要定义一个评分函数,该函数接受特征、对其进行加权,并返回一个数字,以反映每个标签被赋予这些特征和权重的可能性。损失函数将在输入上是凸的:它将返回评分函数的输出与人工提供的标签之间的差异。评分函数的权重矩阵将通过梯度下降进行优化,使损失最小化。
具体来说,我有一些看起来像这样的东西:
entry 1> (ground-truth label), [0.9, 0.2, 1e-2, 6.853, 0.882 ... 1]
*prediction: label 3* *actually: label 4* *loss = some number, update weights*
entry 2> (ground-truth label), [features 1...9]
*prediction: label 1* *actually: label 1* *loss = 0, don't update weights*
...
entry 80,000> (etc...)
如何选择将这些特征映射到五个可能标签之一的可能性的评分函数?
看起来像:
for (loop through all data)
[features] <dot product> [weights] = prediction
if(prediction near (number))
assign label 1
else if (prediction near (number 2)
assign label 2
else (etc...)
hinge_loss_function(prediction) = loss
[weights] = loss*stepsize
end for
这感觉很不对,但我是梯度下降的新手。关于如何进行多标签分类的任何想法?
【问题讨论】:
标签: python machine-learning classification gradient-descent