如何解决keras LSTM多任务学习功能api输入数据
我就像2个数据框一样用于多任务学习,我想预测2个不同数据集的2个值
我收到此错误
ValueError: Data cardinality is ambiguous:
x sizes: 24,41
y sizes: 24,41
Please provide data which shares the same first dimension.
# input tensor
X_data = trainxx.reshape((trainxx.shape[0],1,trainxx.shape[1]))
(24,10)
B_data = trainxx9.reshape((trainxx9.shape[0],trainxx9.shape[1]))
(42,15)
y values are numpy array
ytrainxx
(24,)
ytrainxx9
(42,)
i1 = Input((24,10))
i2 = Input((42,15))
l1 = LSTM(24)(i1)
flat_layer = Flatten()(l1)
l2 = LSTM(42)(i2)
flat_layer2 = Flatten()(l2)
output1 = Dense(1,activation='sigmoid')(flat_layer)
output2 = Dense(1,activation='sigmoid')(flat_layer2)
model = tf.keras.Model(inputs=[i1,i2],outputs=[output1,output2])
model.compile('sgd','mse')
model.fit(x=[X_data,B_data],y=[ytrainxx,ytrainxx9])
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。