我正在尝试使用TensorFlow构建一个用于CIFAR-100数据集的线性分类器。我从Martin Gorner的MNIST教程中获取了代码,并做了一些修改。当我运行这个代码时,TensorFlow并未进行训练(代码在运行,但准确率保持在1.0,损失(交叉熵)保持在4605.17),我不知道哪里出了问题,我实际上是TensorFlow的新手,任何帮助都将不胜感激。
回答:
抱歉发帖了,原来是一个基本的错误。我做了以下更改;
Ytr[0:num_train, Yr[0:num_train]] = 1
Yte[0:num_test, Ye[0:num_test]] = 1
改为
Ytr[range(num_train), Yr_temp[range(num_train)]] = 1 Yte[range(num_test), Ye_temp[range(num_test)]] = 1
前一种方法使得所有值都为1,但我只是想将真实类别的索引设为1,其他元素设为0。感谢您的时间。