tensorflow 2.x中的数据集遇到的问题

问题遇到的现象和发生背景

利用tensorflow中自带的 mnist 数据集遇到的一些问题

问题相关代码，请勿粘贴截图

运行结果及报错内容

数字：9 很容易识别错，这个问题有什么好的解决方法吗

我的解答思路和尝试过的方法

我想要达到的结果https://img-mid.csdnimg.cn/release/static/image/mid/ask/334050130356114.png "#left")

添加部分手写数字9加入到数据集中，重新训练模型