![](/img/trans.png)
[英]TensorFlow/TFLearn: ValueError: Cannot feed value of shape (64,) for Tensor u'target/Y:0', which has shape '(?, 10)'
[英]Tensorflow/TFLearn - ValueError: Cannot feed value of shape (64,) for Tensor 'target/Y:0', which has shape '(?, 1)'
我一直在尝试使用tflearn和我自己的数据集执行深度神经网络来预测值。
我的神经网络以《 泰坦尼克号 》为例,但不同之处是我将“输出”层从2更改为1,并将“ softmax”激活为“线性”:
from tflearn.data_utils import load_csv
data, labels = load_csv('data.csv')
# Build neural network
net = tflearn.input_data(shape=[None, 5])
net = tflearn.fully_connected(net, 5, activation='sigmoid')
net = tflearn.fully_connected(net, 3, activation='sigmoid')
net = tflearn.fully_connected(net, 1, activation='linear')
net = tflearn.regression(net, optimizer='sgd', loss='mean_square', learning_rate=0.1, name='target')
# Define model
model = tflearn.DNN(net)
# Start training (apply gradient descent algorithm)
model.fit(data, labels,show_metric=True)
我收到以下错误:
ValueError:无法为形状为((?,1)'的Tensor'target / Y:0'输入形状(64,)的值
我已经在stackoverflow中搜索了我的问题,但是没有答案对我有用。
我使用Python 3.6和TFlearn 0.3.2
您可以重塑标签
data, labels = load_csv('data.csv')
labels = np.reshape(labels, (-1, 1))
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.