When I run this code:
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers
from tensorflow.keras.datasets import mnist
(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train = x_train.reshape(-1, 784).astype("float32") / 255.0
x_test = x_test.reshape(-1, 784).astype("float32") / 255.0
model = keras.Sequential(
[
layers.Dense(512, activation='relu'),
layers.Dense(256, activation='relu'),
layers.Dense(10),
]
)
model.compile(
loss=keras.losses.SparseCategoricalCrossentropy(from_logits=True),
optimizer=keras.optimizers.Adam(lr=0.001),
metrics=["accuracy"],
The error ModuleNotFoundError: No module named 'keras'
occures due to line model = keras.Sequential...
, however I think that line loss=keras.losses.SparseCategoricalCrossentropy
would give an error too.