MNISTで1だけ取り出す
keras
Lastmod: 2021-07-17

概要

MNISTは言わずと知れた文字画像セットですね。それを使う上で「1」だけ喉から手が出るほどほしい!!!となることがあるかと思います(大袈裟)。そんなときの小技を紹介します。

方法

import keras

(x_train, y_train), (x_test, y_test) = keras.datasets.mnist.load_data()

char = 1 #0でもなんでもOK
x_train_1 = x_train[np.where(y_train==char)]

↑こんだけ。

最後に

(小技でもなんでもない)