更改MNIST Tensorflow的班级数

时间:2017-12-07 01:42:10

标签: tensorflow mnist softmax

嗨,我试图使用MNIST和softmax调整Tensorflow的初学者教程。在本教程中,您有10个clases(数字0-9)。 现在,使用不同的数据集(EMNIST),我有62个数字和字母类。 我在原始模型中的模型是:

x = tf.placeholder(tf.float32, [None, 784])
W = tf.Variable(tf.zeros([784, 10]))
b = tf.Variable(tf.zeros([10]))
y = tf.matmul(x, W) + b`

其中784代表28x28图像的总像素,10代表类的数量。我想要的是:

x = tf.placeholder(tf.float32, [None, 784])
W = tf.Variable(tf.zeros([784, 62]))
b = tf.Variable(tf.zeros([62]))
y = tf.matmul(x, W) + b`

62班。 但是当我到达代码的这一部分时,下一批被调用执行:

  for _ in range(1000):
batch_xs, batch_ys = mnist.train.next_batch(100)
sess.run(train_step, feed_dict={x: batch_xs, y_: batch_ys}) 

我收到这个错误......     回溯(最近一次调用最后一次):

File "calligraphy.py", line 77, in <module>
    tf.app.run(main=main, argv=[sys.argv[0]] + unparsed)
  File "C:\Users\Willy Barales\Anaconda3\lib\site-packages\tensorflow\python\platform\app.py", line 48, in run
    _sys.exit(main(_sys.argv[:1] + flags_passthrough))
  File "calligraphy.py", line 64, in main
    sess.run(train_step, feed_dict={x: batch_xs, y_: batch_ys})
  File "C:\Users\Willy Barales\Anaconda3\lib\site-packages\tensorflow\python\client\session.py", line 789, in run
    run_metadata_ptr)
  File "C:\Users\Willy Barales\Anaconda3\lib\site-packages\tensorflow\python\client\session.py", line 975, in _run
    % (np_val.shape, subfeed_t.name, str(subfeed_t.get_shape())))
ValueError: Cannot feed value of shape (100, 10) for Tensor 'Placeholder_1:0', which has shape '(?, 62)'

有关如何更改此示例的数据集的任何想法? 我是否必须在mnist.py文件中更改实现.next_batch()的内容?

据我所知,EMNIST的格式与MNIST完全相同。 提前致谢。

有关新数据集的信息: http://biometrics.nist.gov/cs_links/EMNIST/Readme.txt

1 个答案:

答案 0 :(得分:0)

我所要做的就是在mnist.py文件中编辑从标签创建一个热矢量的部分,因为那些是与batch_ys相对应的部分,这要归功于Neijla的启发。

def extract_labels(f, one_hot=False, num_classes=62)

当然,除了我在问题中首先说明的,改变模型中类的数量。