2017-01-31 2 views
1

Я тренирую данные, используя приведенный ниже код. Но здесь мне нужно разбить данные на партии (как показано в примере тензорного потока MNIST: batch_xs, batch_ys = mnist.train.next_batch (100)).Разделить данные на партии в Tensorflow для классификации

Когда я искал метод next_batch, он не был доступен ни в одной библиотеке. Я новичок в потоке Тензора. Просто интересно, могу ли я разделить данные на партии во время обучения.

Любая помощь была бы действительно оценена.

for i in range(training_epochs): 
    sess.run(optimizer, feed_dict={x: inputX, y_: inputY}) # Take a gradient descent step using our inputs and labels 

# That's all! The rest of the cell just outputs debug messages. 
# Display logs per epoch step 
    if (i) % display_step == 0: 
    cc = sess.run(cost, feed_dict={x: inputX, y_:inputY}) 
    print "Training step:", '%04d' % (i), "cost=", "{:.9f}".format(cc) #, \"W=", sess.run(W), "b=", sess.run(b) 

print "Optimization Finished!" 
training_cost = sess.run(cost, feed_dict={x: inputX, y_: inputY}) 
print "Training cost=", training_cost, "W=", sess.run(W), "b=", sess.run(b), '\n' 

ответ

0

next_batch не является универсальным библиотечным методом. Это метод, специфичный для набора данных MNIST, который читает следующую мини-сборку входов для подачи te Tensorflow.

Вы можете увидеть его реализацию здесь: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/learn/python/learn/datasets/mnist.py#L160

Вы должны были бы реализовать что-то подобное для набора данных. Этот код не относится к Tensorflow --- вы можете использовать любые библиотеки Python, которые вам нравятся.

Надеюсь, что это поможет!

Смежные вопросы