亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

如何調整 Keras 模型的輸出尺寸?

如何調整 Keras 模型的輸出尺寸?

桃花長相依 2022-06-14 15:13:09
我正在嘗試使用 Keras 構建人工神經網絡。模型的輸入尺寸為 (5, 5, 2),而輸出尺寸為 (5,5)。在運行 keras.fit() 函數時,我遇到以下錯誤:ValueError: Error when checking target: expected dense_3 to have 4 dimensions, but got array with shape (5, 5)這是我正在執行的代碼from keras.models import Sequentialfrom keras.layers import Dense, Flattenimport matplotlib.pyplot as pltfrom keras.callbacks import EarlyStopping, ModelCheckpointmodel = Sequential()model.add(Dense(1000, input_shape=(5, 5, 2), activation="relu"))model.add(Dense(1000, activation="relu"))model.add(Dense(2), output_shape=(5,5))model.summary()model.compile(optimizer="adam",loss="mse", metrics = ["mse"])monitor_val_acc = EarlyStopping(monitor="loss", patience = 10)history = model.fit(trainX, trainYbliss, epochs=1000, validation_data=(testX, testY), callbacks = [monitor_val_acc], verbose = 1)clinical = model.predict(np.arange(0, len(testY)))這是網絡的架構:Layer (type)                 Output Shape              Param #   =================================================================dense_1 (Dense)              (None, 5, 5, 1000)        3000      _________________________________________________________________dense_2 (Dense)              (None, 5, 5, 1000)        1001000   _________________________________________________________________dense_3 (Dense)              (None, 5, 5, 1)           1001      =================================================================Total params: 1,005,001Trainable params: 1,005,001Non-trainable params: 0_________________________________________________________________模型應該基于 (5,5,2) 數組輸出 (5,5) 數組,但在最低隱藏層失敗。我該如何解決這個問題?
查看完整描述

2 回答

?
開滿天機

TA貢獻1786條經驗 獲得超13個贊

使用下面的代碼作為參考根據您的輸入值更改值:

train_data = train_data.reshape(train_data.shape[0], 10, 30, 30, 1)

對于您的輸入火車數據,



查看完整回答
反對 回復 2022-06-14
?
慕尼黑8549860

TA貢獻1818條經驗 獲得超11個贊

您的網絡將輸出一個 shape 的張量(batch_size, 5, 5, 1)。您的輸出是 4 維張量嗎?如果它是一個單一的價值,(5,5)你需要將它重塑成(1,5,5,1)我認為



查看完整回答
反對 回復 2022-06-14
  • 2 回答
  • 0 關注
  • 237 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號