class_pic / load_dataset.py
the10or's picture
Upload load_dataset.py
a2188b7 verified
raw
history blame contribute delete
341 Bytes
from tensorflow import keras
from tensorflow import keras
# Load CIFAR-10 dataset
(x_train, y_train), (x_test, y_test) = keras.datasets.cifar10.load_data()
# Define class names
classes = [
"airplane",
"automobile",
"bird",
"cat",
"deer",
"dog",
"frog",
"horse",
"ship",
"truck"
]