多标签分类的预训练

Question

我必须预先训练一个用于多标签分类的模型。我正在使用cifar10数据集进行预训练，我想知道是否必须使用预训练 'categorical_crossentrpy'（softmax）或'binary_crossentropy'（Sigmoid），因为在第一种情况下，我遇到了多分类问题

Answer 1

您应该使用wb = load_workbook(in_mem_file, read_only=True, data_only=True) ws = wb[sheet_name] ，因为它为您提供了每个班级的概率，无论那里有多少个概率。正如您所写的那样，Sigmoid与softmax一起使用，并用于二进制分类（因此名称为binnary）。我希望现在更清楚。