小编she*_*rry的帖子

我目前正在使用Keras在MNIST数据集上训练前馈神经网络。我正在使用以下格式加载数据集

(X_train, Y_train), (X_test, Y_test) = mnist.load_data()

但是我只想用数字0和4训练我的模型，而不是全部。如何只选择2位数字？我对python相当陌生，可以弄清楚如何过滤mnist数据集...