0
這是一個簡單的神經網絡,包含3個輸入值和3個輸出值。keras神經網絡架構不正確
錯誤:
ValueError: Error when checking model target: expected dense_78 to have shape (None, 3) but got array with shape (3, 1)
是,當我執行此網絡拋出。我已經設定的最後一層有匹配的標籤數這3個可能的輸出:
model.add(Dense(3, activation='softmax'))
我沒有正確的架構這個網絡,哪裏是我的錯?
data = ([[ 0.29365378],
[ 0.27958957],
[ 0.27946938]])
labels = [[1], [2], [3]]
import numpy as np
from keras.models import Sequential
from keras.layers import Dense, Dropout, Activation
from keras.optimizers import SGD
model.add(Dense(64, activation='relu', input_dim=1))
model.add(Dropout(0.5))
model.add(Dense(64, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(3, activation='softmax'))
sgd = SGD(lr=0.01, decay=1e-6, momentum=0.9, nesterov=True)
model.compile(loss='categorical_crossentropy',
optimizer=sgd,
metrics=['accuracy'])
model.fit(data, labels,
epochs=20,
batch_size=32)