import tensorflow as tf
from tensorflow import keras
import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"
(x_train, y_train), (x_test, y_test) = keras.datasets.mnist.load_data()
x_train = x_train / 255.0
x_test = x_test / 255.0
model = keras.Sequential([
keras.layers.Flatten(input_shape=(28,28)),
keras.layers.Dense(128, activation='relu'),
keras.layers.Dropout(0.2),
keras.layers.Dense(10),
])
model. compile(optimizer='adam', loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True), metrics=['accuracy'])
model.fit(x_train, y_train, epochs=10, validation_data=(x_test, y_test))
test_loss, test_acc = model.evaluate(x_test, y_test, verbose=2)
print('Test accuracy:', test_acc)
aW1wb3J0IHRlbnNvcmZsb3cgYXMgdGYgCmZyb20gdGVuc29yZmxvdyBpbXBvcnQga2VyYXMKCmltcG9ydCBvcwpvcy5lbnZpcm9uWyJDVURBX1ZJU0lCTEVfREVWSUNFUyJdPSItMSIKCih4X3RyYWluLCB5X3RyYWluKSwgKHhfdGVzdCwgeV90ZXN0KSA9IGtlcmFzLmRhdGFzZXRzLm1uaXN0LmxvYWRfZGF0YSgpCgp4X3RyYWluID0geF90cmFpbiAvIDI1NS4wCnhfdGVzdCA9IHhfdGVzdCAvIDI1NS4wCgptb2RlbCA9IGtlcmFzLlNlcXVlbnRpYWwoWwprZXJhcy5sYXllcnMuRmxhdHRlbihpbnB1dF9zaGFwZT0oMjgsMjgpKSwKa2VyYXMubGF5ZXJzLkRlbnNlKDEyOCwgYWN0aXZhdGlvbj0ncmVsdScpLAprZXJhcy5sYXllcnMuRHJvcG91dCgwLjIpLCAKa2VyYXMubGF5ZXJzLkRlbnNlKDEwKSwKXSkKCm1vZGVsLiBjb21waWxlKG9wdGltaXplcj0nYWRhbScsIGxvc3M9dGYua2VyYXMubG9zc2VzLlNwYXJzZUNhdGVnb3JpY2FsQ3Jvc3NlbnRyb3B5KGZyb21fbG9naXRzPVRydWUpLCBtZXRyaWNzPVsnYWNjdXJhY3knXSkKCm1vZGVsLmZpdCh4X3RyYWluLCB5X3RyYWluLCBlcG9jaHM9MTAsIHZhbGlkYXRpb25fZGF0YT0oeF90ZXN0LCB5X3Rlc3QpKQoKdGVzdF9sb3NzLCB0ZXN0X2FjYyA9IG1vZGVsLmV2YWx1YXRlKHhfdGVzdCwgeV90ZXN0LCB2ZXJib3NlPTIpCnByaW50KCdUZXN0IGFjY3VyYWN5OicsIHRlc3RfYWNjKSA=