我正在尝试接受 2 个输入到我的模型中。但它有一个奇怪的问题。
x1= layers.Input((20000,))
x2= layers.Reshape((200,100), input_shape=(20000,))(x1)
y1= layers.Input((200000,))
y2= layers.Reshape((2000,100), input_shape=(200000,))(y1)
combine = layers.Concatenate(axis=1)([x2, y2])
model = tf.keras.Model(inputs=[x1, y1], outputs=combine)
model.predict([datasetA, datasetB])
如果我接受一个输入,模型就可以运行。
model = tf.keras.Model(inputs=[x1], output=x2)
model.predict(datasetA)
但如果我接受两个输入,模型就死了。
Failed to find data adapter that can handle input: (<class 'list'> containing values of types {"<class 'tensorflow.python.data.ops.dataset_ops.PrefetchDataset'>"}), <class 'NoneType'>
即我的数据集的结构是:
<PrefetchDataset shapes: ((None, 20000), (None,)), types: (tf.int64, tf.int32)>
1 Dataset -> 5 Batche_data and 5 batch_Label
Each Batch_data -> 3 records
Each record -> [20000]
Each batch_Label -> 3 records
Each label -> 0 / 1 for classification
我该如何解决这个问题?
扬帆大鱼
慕哥9229398