Несовместимые формы: [14250,10] против [15000,10] - PullRequest
Новая
       9

Несовместимые формы: [14250,10] против [15000,10]

0 голосов
/ 22 сентября 2018

Я новичок в tenorflow и слежу за программой udacity.Меня попросили построить классификатор изображений на основе набора данных notMNIST.

Я прилагаю код и ошибку, пожалуйста, помогите мне с любыми возможными способами.Спасибо. 

features_count = 784
labels_count = 10

features = tf.placeholder(tf.float32)
labels = tf.placeholder(tf.float32)

weights = tf.Variable(tf.truncated_normal((features_count, 
labels_count)))
biases = tf.Variable(tf.zeros(labels_count))

train_feed_dict = {features: train_features,labels: train_labels}
valid_feed_dict = {features: valid_features, labels: valid_labels}
test_feed_dict = {features: test_features, labels: test_labels}

# Linear function Wx+b
logits = tf.add(tf.matmul(features,weights), biases)

prediction = tf.nn.softmax(logits)

#Cross entropy
cross_entropy = -tf.reduce_sum(labels * tf.log(prediction), 
reduction_indices = 1)

# Training loss
loss = tf.reduce_mean(cross_entropy)

# Initialize all the variables
init = tf.global_variables_initializer()

with tf.Session() as sess:
  sess.run(init)
  sess.run(loss, feed_dict = train_feed_dict)
  sess.run(loss, feed_dict = valid_feed_dict)
  sess.run(loss, feed_dict = test_feed_dict)
  biases_data = sess.run(biases)

Это ошибка, которую я получаю: 

InvalidArgumentError: Incompatible shapes: [14250,10] vs. [15000,10]
 [[Node: mul_7 = Mul[T=DT_FLOAT, 
_device="/job:localhost/replica:0/task:0/device:CPU:0"] . 
(_arg_Placeholder_16_0_1, Log_7)]]

During handling of the above exception, another exception occurred:

InvalidArgumentError                      Traceback (most recent call 
last)
<ipython-input-46-83a463c8a201> in <module>()
 53 with tf.Session() as sess:
 54     sess.run(init)
 ---> 55     sess.run(loss, feed_dict = train_feed_dict)
 56     sess.run(loss, feed_dict = valid_feed_dict)
 57     sess.run(loss, feed_dict = test_feed_dict)
...