神经网络的输出答案

【问题标题】：Output of a neural network神经网络的输出
【发布时间】：2020-03-15 16:46:48
【问题描述】：

我使用 mnist 数据集测试我的网络。因此模型的输出形状为 10。

如何重塑输出？例如，如果输出是标签3，那么输出是[0 0 0 1 0 0 0 0 0 0]，还是[0 0 0 3 0 0 0 0 0 0]，还是完全不同？

问题是我不想使用数据加载器。我用这个方法：

from mlxtend.data import loadlocal_mnist
X, y = loadlocal_mnist(
            images_path='/home/wai043/data/mnist/train-images-idx3-ubyte', 
            labels_path='/home/wai043/data/mnist/train-labels-idx1-ubyte')

【问题讨论】：

这里的输出是什么，是你执行分类后的结果还是你从mnist加载后得到的结果？

标签： python neural-network pytorch mnist

【解决方案1】：

如果标签为 3，则输出必须为 [0, 0, 0, 1, 0, 0, 0, 0, 0, 0]。您从 loadlocal_mnist 获得的 y 参数具有直接标签，因此您需要在训练之前“一次性编码”y。

您可以使用以下代码进行编码

from mlxtend.preprocessing import one_hot
from mlxtend.data import loadlocal_mnist

X, y = loadlocal_mnist(images_path='/home/wai043/data/mnist/train-images-idx3-ubyte', 
                       labels_path='/home/wai043/data/mnist/train-labels-idx1-ubyte')
y = one_hot(y)

【讨论】：