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]) 

但我得到的形狀（1，1，54）的一個不能喂值張量u'InputData/X：0' ，其具有形狀「（54？）。

我知道我必須重塑它，但並不確定如何從我在網上看到的其他示例去了解它。

Answer 1

它看起來像你包裹在一個額外的列表。嘗試pred = model.predict(df)。