我正在嘗試運(yùn)行一個(gè)簡單的神經(jīng)網(wǎng)絡(luò),并且使用以下代碼已經(jīng)達(dá)到了我的特征平坦的地步:training_dataset = (tf.data.Dataset.from_tensor_slices( ( tf.cast(ballast_train[features].values, tf.float64), tf.cast(ballast_train[target].values, tf.int32) )))for features_tensor, target_tensor in training_dataset: print(f'features:{features_tensor} target:{target_tensor}')features:[0.46029711 0.33290338 0.78302964 0.10295655 0.5890411 ] target:5features:[0.63530873 0.90712946 0.27781778 0.10295655 0.45988258] target:5features:[0.68413444 0.81390713 0.8448272 0.65073914 0.46771037] target:2現(xiàn)在,我嘗試運(yùn)行以下代碼,但無法正確獲取代碼的 tf.keras.Input() 部分。`inputs = tf.keras.Input(shape=(5,))x = tf.keras.layers.Dense(100, activation=tf.nn.relu)(inputs)outputs = tf.keras.layers.Dense(15, activation=tf.nn.softmax)(x)model = tf.keras.Model(inputs=inputs, outputs=outputs)model.compile(optimizer='adam', loss='mean_squared_error', metrics=['accuracy'])model.fit(training_dataset, epochs=5)`當(dāng)嘗試擬合模型時(shí),會出現(xiàn)此錯(cuò)誤:ValueError: Error when checking input: expected input_10 to have shape (5,) but got array with shape (1,)“形狀”參數(shù)中應(yīng)該包含什么?我在這里缺少什么嗎?
1 回答

當(dāng)年話下
TA貢獻(xiàn)1890條經(jīng)驗(yàn) 獲得超9個(gè)贊
嘗試以下方法看看是否有效:
model.fit(training_dataset.batch(batch_size),?epoch=5)
您需要提供一個(gè)值batch_size
如果這不起作用,我總是傾向于將我的功能和示例提供給單獨(dú)的參數(shù)x
和y
tf.keras.Model.fit?。您可以嘗試修改代碼來執(zhí)行此操作,而不是將它們合并到數(shù)據(jù)集中。
添加回答
舉報(bào)
0/150
提交
取消