0

我来自 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])

但是我得到了一个无法输入形状 (1, 1, 54) 的 Tensor u'InputData/X:0' 值，它的形状为 '(?, 54)。

我知道我必须重塑它，但不确定如何从我在网上看到的其他示例中确切地去做。

4

1 回答 1

1

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

于 2016-12-10T18:36:28.753 回答