2016-12-10 4 views
0

Я иду из SciKit и пытаюсь получить винт TensorFlow.TensorFlow/TFLearn не может задать значение формы

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) 

Мое приложение затем проходит через петлю и имеет функцию, которая принимает dataframe, преобразует его в массив нп и пытается сделать прогноз на него.

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

но я получаю не может кормить значение формы (1, 1, 54) для тензорной u'InputData/Х: 0' , который имеет форму «(54?).

Я понимаю, что мне нужно изменить его, но не совсем точно, как это сделать из других примеров, которые я видел в Интернете.

ответ

1

Похоже, вы заверните его в дополнительный список. Попробуйте pred = model.predict(df).