TensorFlow
小智 5
因为我看不到你的 trainY 编码;看起来 - 你的 trainY 只有一列,你的模型输出有 10 个神经元,所以 Shapes (None, 1) 和 (None, 10) 是不兼容的。你可以在你的 trainY 上试试这个(即 one-hot 编码)
from sklearn.preprocessing import LabelBinarizerlabel_as_binary = LabelBinarizer()train__y_labels = label_as_binary.fit_transform(trainY)编译看起来像(寻找train__y_labels)
model.pile(optimizer='adam', loss='categorical_crossentropy', metrics=['auracy'])model.fit(train_X_input, train__y_labels, batch_size=128, epochs=1)注意:如果您的 valid 也引发错误,则所有 y(s) 都需要相同。
TensorFlow