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个输出。