0
Tensorflowを使用するMNISTデータセット用のPythonコードがあります。独自のcsvファイルを使用してバッチを作成するpython
with tf.Session() as sess:
sess.run(tf.initialize_all_variables())
for epoch in range(hm_epochs):
epoch_loss = 0
for _ in range(int(mnist.train.num_examples/batch_size)):
epoch_x, epoch_y = mnist.train.next_batch(batch_size)
_, c = sess.run([optimizer, cost], feed_dict={x: epoch_x, y: epoch_y})
epoch_loss += c
print('Epoch: ', epoch, ' completed out of: ', hm_epochs, ' loss: ', epoch_loss)
correct = tf.equal(tf.argmax(prediction, 1), tf.argmax(y, 1))
accuracy = tf.reduce_mean(tf.cast(correct, 'float'))
print('Accuracy:', accuracy.eval({x: mnist.test.images, y: mnist.test.labels}))
ライン:
epoch_x, epoch_y = mnist.train.next_batch(batch_size)
は100
私の質問は、新しい行に次の行を置換する方法、ある大きさで毎回新しいバッチを作っている セッションは以下のとおりです。自分のCSVファイル(リストのリスト)があれば、それは私の新しいバッチになりますか? 私の現在のコードは次のようになります。
「nextbatchは」私は定義された関数があるwith tf.Session() as sess:
sess.run(tf.initialize_all_variables())
for epoch in range(hm_epochs):
epoch_loss = 0
for _ in range(len(training_data_list) // batch_size):
epoch_x, epoch_y = training_data_list.nextbatch(batch_size)
_, c = sess.run([optimizer, cost], feed_dict={x: epoch_x, y: epoch_y})
epoch_loss += c
print('Epoch: ', epoch, ' completed out of: ', hm_epochs, ' loss: ', epoch_loss)
correct = tf.equal(tf.argmax(prediction, 1), tf.argmax(y, 1))
accuracy = tf.reduce_mean(tf.cast(correct, 'float'))
print('Accuracy:', accuracy.eval({x: inputs, y: targets}))
。私はどんな提案に感謝
AttributeError: 'list' object has no attribute 'nextbatch'
:しかし、私は次のエラーを取得するあなたが扱うオブジェクトを実装する必要が
stops = open('.../Desktop/stops_train.csv', 'r')
training_data_list = stops.readlines()
stops.close()
ありがとうございます。 一方、私はこれを使用しました: _範囲(_en * batch_size: batch_size] そしてそれは働いた。 – CrisH