HDF5数据Caffe中的多级分类

Question

我有900个一维数据作为训练数据和100个一维数据作为测试数据。我将我的数据存储为.hdf5文件。 对于二元分类（例如0和1），我使用Caffe，对结果非常满意。

但是现在我需要在Caffe中为多类分类（例如1,2,3,4类）构建一个深度学习模型。任何人都可以请帮我解释一下，我如何训练Caffe网络进行多级分类？

Answer 1

对于多类问题，可以使用相同的SoftmaxWithLoss。唯一的区别是，在最后一层，每个类应该有一个通道，标签数据应该编号为0,1，... C-1，其中C是类的数量。你可以在alexnet中查看一个例子，他们在那里训练1000个班的模型（ImageNet）。你可以看到在最后一层（丢失之前）它们有1000个输出。