TensorFlow错误:ValueError:形状(128,100)和(128,100,139)不兼容

发布于 2025-01-26 06:17:25 字数 1025 浏览 2 评论 0原文

我尝试将功能API用于模型,但我不明白为什么有错误

ValueError: Shapes (128, 100) and (128, 100, 139) are incompatible

input_tensor = Input(batch_input_shape=(batch_size,None))
x = Embedding(vocab_size, embed_dim)(input_tensor)
x = LSTM(rnn_neurons4, return_sequences=True, stateful=True)(x)
output_tensor = Dense(vocab_size, activation='softmax')(x)
model = Model(input_tensor, output_tensor) 

model.summary()

Adam = tf.keras.optimizers.Adam(learning_rate=0.0001)
model.compile(optimizer=Adam, loss="categorical_crossentropy", metrics=['accuracy'])

“

适合代码:

epochs = 1000
early_stop = EarlyStopping(monitor='loss', patience=25)
try:
  model.fit(dataset,epochs=epochs, callbacks=[early_stop])
  model.save('train.h5')
except KeyboardInterrupt:
  model.save('train.h5')

I try to use Functional API for my model, but i don't understand why i have error:

ValueError: Shapes (128, 100) and (128, 100, 139) are incompatible

My code:

input_tensor = Input(batch_input_shape=(batch_size,None))
x = Embedding(vocab_size, embed_dim)(input_tensor)
x = LSTM(rnn_neurons4, return_sequences=True, stateful=True)(x)
output_tensor = Dense(vocab_size, activation='softmax')(x)
model = Model(input_tensor, output_tensor) 

model.summary()

Adam = tf.keras.optimizers.Adam(learning_rate=0.0001)
model.compile(optimizer=Adam, loss="categorical_crossentropy", metrics=['accuracy'])

model summory

fit code:

epochs = 1000
early_stop = EarlyStopping(monitor='loss', patience=25)
try:
  model.fit(dataset,epochs=epochs, callbacks=[early_stop])
  model.save('train.h5')
except KeyboardInterrupt:
  model.save('train.h5')

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

评论(1

失与倦" 2025-02-02 06:17:25

我使用sparse_categorical_crossential创建自己的功能,然后添加型号。

I create my own function with sparse_categorical_crossential and add in model.compile

~没有更多了~
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文