Я иду из 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?).
Я понимаю, что мне нужно изменить его, но не совсем точно, как это сделать из других примеров, которые я видел в Интернете.