问函数mnist.train.next_batch()在训练数据集中的用途是什么？
EN

Stack Overflow用户

提问于 2019-01-02 14:46:02

回答 1查看 448关注 0票数 0

我正在使用TensorFlow通过胶囊网络训练我自己的数据集。在训练mnist数据集时，它包含函数mnist.train.next_batch(批量大小)。如何用TensorFlow替换这个函数来训练自己的数据集？

发布于 2019-01-02 14:50:21

函数sample batch_size从混洗后的训练数据集中获取样本数量，然后返回批次进行训练。

您可以编写自己的next_batch()方法来做同样的事情，也可以根据需要对其进行修改。然后在训练模型时以类似的方式使用它。

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/54002301

复制

相似问题

问函数mnist.train.next_batch()在训练数据集中的用途是什么？EN