python - Keras CNN 中的标签预处理

Question

我有一个 600 行和 271 列的数据集（最后一列包含类）。每行包含恶意软件特征及其类别标签。我正在编写卷积神经网络代码来预测这些类。共有9个班。下面是我的代码：

 dataset = pd.read_csv('train.csv')
 X = dataset.iloc[:, 0:270].values
 y = dataset.iloc[:, 270].values
 from sklearn.preprocessing import LabelEncoder, OneHotEncoder
 from sklearn.model_selection import train_test_split
 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.20)
 model = Sequential()
 model.add(Convolution1D(64, 10, input_shape=(643,270)))
 model.add(Activation('relu'))
 model.add(MaxPooling1D(1))
 model.add(Flatten())
 model.add(Dense(100))
 model.add(Dropout(0.5))
 model.add(Dense(9))
 model.add(Activation('softmax'))
 model.compile(loss='categorical_crossentropy', optimizer = 'adam', metrics = ['accuracy'])
 X_train = np.reshape(X_train, (1, X_train.shape[0], X_train.shape[1]))
 X_test = np.reshape(X_test, (1, X_test.shape[0], X_test.shape[1]))
 y_train = np_utils.to_categorical(y_train, 10) 
 y_test = np_utils.to_categorical(y_test, 10)
 model.fit(X_train,y_train,validation_data=(X_test,y_test))
 print(str(model.evaluate(x_test,y_test)))

在这里，我在此行出现错误：

 model.fit(X_train,y_train,validation_data=(X_test,y_test))
 Error: Error when checking target: expected activation_2 to have shape (None, 9) 
 but got array with shape (643, 10)

因为“y_train”的形状不适合训练。有人可以解释我如何解决这个问题并重塑 y_train 以进行成功的培训吗？

编辑： 更改input_shape为第二和第三维形状后，错误现在更改为：

 fValueError: Error when checking target: expected activation_2 to have shape (None, 1) 
 but got array with shape (643, 10) 

请问这个问题有什么解决办法吗？谢谢！

标签： pythontensorflowneural-networkkerasconv-neural-network