在数据集中,每个标签都可能被命名(例如:'rock')。要将其与神经网络一起使用,需要将其转换为整数(例如:2),然后转换为单热编码(例如:[0,0,1])。所以'rock' == 2 == [0,0,1]。您的输出预测将采用这种单热编码形式。 [ 0.1, 0.1, 0.9 ] 表示预测了第 2 类,[ 0.9, 0.1, 0.1 ] 表示第 0 类等。
要以可逆方式执行此操作,请使用sklearn.preprocessing.LabelBinarizer。
有几种方法可以将帧预测组合成整体预测。最常见的是,按照复杂性递增的顺序:
- 对概率的多数投票
- 对概率的平均/平均投票
- 对概率的对数赔率进行平均
- 概率对数概率的序列模型
- 多实例学习
以下是前三个的示例。
import numpy
from sklearn.preprocessing import LabelBinarizer
labels = [ 'rock', 'jazz', 'blues', 'metal' ]
binarizer = LabelBinarizer()
y = binarizer.fit_transform(labels)
print('labels\n', '\n'.join(labels))
print('y\n', y)
# Outputs from frame-based classifier.
# input would be all the frames in one song
# frame_predictions = model.predict(frames)
frame_predictions = numpy.array([
[ 0.5, 0.2, 0.3, 0.9 ],
[ 0.9, 0.2, 0.3, 0.3 ],
[ 0.5, 0.2, 0.3, 0.7 ],
[ 0.1, 0.2, 0.3, 0.5 ],
[ 0.9, 0.2, 0.3, 0.4 ],
])
def vote_majority(p):
voted = numpy.bincount(numpy.argmax(p, axis=1))
normalized = voted / p.shape[0]
return normalized
def vote_average(p):
return numpy.mean(p, axis=0)
def vote_average_logits(p):
logits = numpy.log(p / (1 - p))
avg = numpy.mean(logits, axis=1)
p = 1/(1+ numpy.exp(-avg))
return p
maj = vote_majority(frame_predictions)
mean = vote_average(frame_predictions)
mean_logits = vote_average_logits(frame_predictions)
genre_maj = binarizer.inverse_transform(numpy.array([maj]))
genre_mean = binarizer.inverse_transform(numpy.array([mean]))
genre_mean_logits = binarizer.inverse_transform(numpy.array([mean_logits]))
print('majority voting', maj, genre_maj)
print('mean voting', mean, genre_mean)
print('mean logits voting', mean_logits, genre_mean_logits)
输出
labels:
rock
jazz
blues
metal
y:
[[0 0 0 1]
[0 1 0 0]
[1 0 0 0]
[0 0 1 0]]
majority voting: [0.4 0. 0. 0.6] ['rock']
mean voting: [0.58 0.2 0.3 0.56] ['blues']
mean logits voting [0.49772704 0.44499443 0.41421356 0.24829914 0.4724135 ] ['blues']
对平均概率的一个简单改进是计算概率的 logits(log-odds)并将其平均。这更恰当地解释了很可能或不太可能发生的事情。它可以看作是一个朴素贝叶斯,在帧是独立的假设下计算后验概率。
也可以通过使用在逐帧预测上训练的分类器来执行投票,尽管这并不常见,并且当输入长度变化时会很复杂。可以使用简单的序列模型,即循环神经网络 (RNN) 或隐马尔可夫模型 (HMM)。
另一种选择是在基于帧的分类上使用Multiple-Instance-Learning with GlobalAveragePooling,一次性学习整首歌曲。