特定のクラス名に基づいてCIFAR-10データセットを読み込む方法は？

Question

私は深い学習のためにCIFAR-10データセットを使用していますが、フルーツクラスのデータセットのみを指定したいと思います。

(X_train, y_train), (X_test, y_test) = cifar10.load_data() 

すべてのCIFAR-10データセットをロードすることがわかりました。データのすべてではなくフルーツクラスのデータのみを読み込む方法は？

Answer 1

追加データを読み込んでも構わない場合は、魔法使いが果物のラベルであることを確認し、X_train, y_train = X_train[y_train == fruit_label], y_train[y_train == fruit_label]のようにして、データがnp.arraysに保存されていることを前提にしてください。あなたのテストセットに相当します。

もしそうでなければ、hdf5ファイルを変更する必要があります。