Keras 多标签分类 'to_categorical' 错误

Question

接收中

IndexError: index 3 is out of bounds for axis 1 with size 3

尝试在输出向量上使用 Keras to_categorical 创建单热编码时。 Y.shape = (178,1)。请帮忙(:

import keras
from keras.models import Sequential
from keras.layers import Dense
import numpy as np

# number of wine classes
classifications = 3

# load dataset
dataset = np.loadtxt('wine.csv', delimiter=",")
X = dataset[:,1:14]
Y = dataset[:,0:1]

# convert output values to one-hot
Y = keras.utils.to_categorical(Y, classifications)

# creating model
model = Sequential()
model.add(Dense(10, input_dim=13, activation='relu'))
model.add(Dense(15, activation='relu'))
model.add(Dense(20, activation='relu'))
model.add(Dense(classifications, activation='softmax'))

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

model.fit(X, Y, batch_size=10, epochs=10)

Answer 1

好吧，问题在于 wine 标签来自范围 [1, 3] 和 to_categorical 索引 class 来自 0。当将 3 标记为 to_categorical 时会出错，将此索引视为实际的第 4 个 class - 这与您提供的 classes 的数量不一致。最简单的解决方法是通过以下方式枚举从 0 开始的标签：

Y = Y - 1

Keras 多标签分类 'to_categorical' 错误

Keras Multi-Label Classification 'to_categorical' Error

machine-learning

neural-network

keras

multilabel-classification