如何使用 Keras 进行多类分类？

Question

我想用 Keras 制作简单的分类器来对我的数据进行分类。特征是数字数据，结果是字符串/分类数据。我预测有 15 个不同的类别/类别。这就是我的代码的样子：

model = Sequential()
model.add(Dense(16, input_dim = x_train.shape[1], activation = 'relu')) # input layer requires input_dim param
model.add(Dense(16, activation = 'relu'))
model.add(Dense(16, activation = 'relu'))
model.add(Dense(1, activation='relu'))

model.compile(loss="binary_crossentropy", optimizer= "adam", metrics=['accuracy'])

#es = EarlyStopping(monitor='loss', min_delta=0.005, patience=1, verbose=1, mode='auto')
model.fit(x_train, y_train, epochs = 100, shuffle = True, batch_size=128, verbose=2)

scores = model.evaluate(x_test, y_test)
print(model.metrics_names[0], model.metrics_names[1])

问题是我总是收到此错误：

ValueError: could not convert string to float 'category1'

我究竟做错了什么？

当我用整数替换我的类名称“category1”、“category2”等时，我的代码可以工作，但它总是给我 0 的精度。我尝试更改节点数、层数和激活函数，但结果始终为 0。看起来模型认为我正在做回归而不是分类。

如果我的分类值不只是 1 或 0，使用 Keras lib 进行分类的正确方法是什么？

Answer 1

您需要将字符串类别转换为整数，有一个方法可以实现：

y_train = tf.keras.utils.to_categorical(y_train, num_classes=num_classes)

此外，多类分类的最后一层应该是这样的：

model.add(Dense(NUM_CLASSES, activation='softmax'))

最后，对于多类分类，正确的损失将是类别交叉熵。

model.compile(loss="categorical_crossentropy", optimizer= "adam", metrics=['accuracy'])

这是张量流中提供的一个很好的示例：分类示例