diff --git a/m1_tf_test.py b/m1_tf_test.py new file mode 100644 index 0000000..df65585 --- /dev/null +++ b/m1_tf_test.py @@ -0,0 +1,25 @@ +#!/usr/bin/env python3 +# -*- coding:utf-8 -*- +############################################################# +# File: m1_test.py +# Created Date: Saturday July 9th 2022 +# Author: Smiril +# Email: sonar@gmx.com +############################################################# +import tensorflow as tf +tf.__version__ +tf.config.list_physical_devices() +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) +]) +loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True) +model.compile(optimizer='adam', + loss=loss_fn, + metrics=['accuracy']) +model.fit(x_train, y_train, epochs=10)