[英]error in prediction - Keras Functional API
我已經訓練了這個模型:
from keras.layers import concatenate
from keras.layers import Input, Dense, Masking
from keras.models import Model
x_in = Input(shape=(5,), name='x_in')
s_in = Input(shape=(18,), name='s_in')
s_masked = Masking(0.0)(s_in)
z = concatenate([x_in,s_masked])
dense_1 = Dense(40, kernel_initializer='normal', activation='relu', name='dense_1')(z)
dense_2 = Dense(40, kernel_initializer='normal', activation='relu', name='dense_2')(dense_1)
output = Dense(1, kernel_initializer='normal', name = 'output')(dense_2)
model = Model(inputs=[x_in,s_in], outputs=output)
model.compile(optimizer='adam', loss={'output':'mean_squared_error'})
model.fit({'x_in': x_training,'s_in':s_training},{'output':y_training},batch_size=30, epochs=10, validation_split=0.3, shuffle=True, callbacks=[plot_losses])
我現在想現在預測。 但是由於我有多個輸入,因此我不知道如何使用model.predict()
。
當我嘗試:
predictions = model.predict(x_testing, s_testing)
print predictions
我收到此錯誤:
該模型需要2個輸入數組,但只接收一個數組。 找到:具有形狀的數組(1710,5)
我不明白,因為我給了兩個數組x_testing
和s_testing
您應該像建立模型時所做的那樣,將輸入作為級聯數組提供 ( inputs=[x_in,s_in]
):
predictions = model.predict([x_testing, s_testing])
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.