commit e5a3bef0ae87c6f0808a46d09520aff161f028e4 Author: Bart Moyaers Date: Thu Dec 12 11:28:00 2019 +0100 add first simple TF test diff --git a/test1.py b/test1.py new file mode 100644 index 0000000..4172fc4 --- /dev/null +++ b/test1.py @@ -0,0 +1,19 @@ +import tensorflow as tf +mnist = tf.keras.datasets.mnist + +(x_train, y_train),(x_test, y_test) = mnist.load_data() +x_train, x_test = x_train / 255.0, x_test / 255.0 + +model = tf.keras.models.Sequential([ + tf.keras.layers.Flatten(input_shape=(28, 28)), + tf.keras.layers.Dense(128, activation='relu'), + tf.keras.layers.Dropout(0.2), + tf.keras.layers.Dense(10, activation='softmax') +]) + +model.compile(optimizer='adam', + loss='sparse_categorical_crossentropy', + metrics=['accuracy']) + +model.fit(x_train, y_train, epochs=5) +model.evaluate(x_test, y_test) \ No newline at end of file