TensorFlow/TFLearn 无法提供形状值

Question

我来自 SciKit，正在尝试掌握 TensorFlow 的窍门。

net = tflearn.input_data(shape=[None, 54])
net = tflearn.fully_connected(net, 32)
net = tflearn.fully_connected(net, 32)
net = tflearn.fully_connected(net, 32)
net = tflearn.fully_connected(net, 2, activation='softmax')
net = tflearn.regression(net)
model = tflearn.DNN(net)
model.fit(data, labels, n_epoch=10, batch_size=1, show_metric=True)

我的应用程序然后通过一个循环运行并具有一个函数，该函数接受数据帧，将其转换为 np 数组并尝试对其进行预测。

def predict(predict_dataframe):
    df = predict_dataframe.values
    pred = model.predict([df])

但是我得到了张量 u'InputData/X:0' 的形状为 (1, 1, 54) 的无法馈送值，其形状为 '(?, 54).

我知道我必须重塑它，但不确定如何根据我在网上看到的其他示例进行重塑。

Answer 1

您似乎将其包装在一个额外的列表中。试试 pred = model.predict(df).

TensorFlow/TFLearn 无法提供形状值

TensorFlow/TFLearn Cannot Feed Value Of Shape

tensorflow

tflearn