我开始在Node.js中使用一个非常基本的Deep Belief Network,但它不够快。基本上它使用X
和Y
,其中每个都是一个数组数组; X
是要训练的数据,Y
就是结果。
所以我会提供类似var x=[[1,2,3], [1,3,2]]
等的东西。等等。和y=[[1,0], [1,0]]
。然后我会提供一些数据,例如[2,3,1]
,它会预测y
。
我在tfslearn中失去了如何做到这一点。我可以自己学习,但我已经达到了一个我甚至不确定谷歌甚至是什么的地步。
如果它只是一个数组,我可以使用这些示例。
每当我尝试使用数组数组时,我得到:
无法提供形状值
答案 0 :(得分:0)
我正在为我的数据集错误地设置输入形状。这有很大帮助:http://tflearn.org/tutorials/quickstart.html
# Data loading and preprocessing
# Building deep neural network
net = tflearn.input_data(shape=[None, 4])
net = tflearn.fully_connected(net, 32)
net = tflearn.fully_connected(net, 32)
net = tflearn.fully_connected(net, 1, activation='softmax')
net = tflearn.regression(net)
# Training
model = tflearn.DNN(net)
model.fit(X, Y, n_epoch=10, batch_size=16, show_metric=True)