From 12eaf8733579b64d3bcfe6e8a40afc5a06103d3f Mon Sep 17 00:00:00 2001 From: Smiril Date: Sun, 10 Jul 2022 22:58:22 +0200 Subject: [PATCH] Update m1_tf_test.py --- m1_tf_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/m1_tf_test.py b/m1_tf_test.py index 41a7382..2ac8ba1 100644 --- a/m1_tf_test.py +++ b/m1_tf_test.py @@ -32,8 +32,8 @@ loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True) model.compile(optimizer='adam', loss=loss_fn, metrics=['accuracy']) -model.fit(x_test, y_test, epochs=10) -outputs = tf.keras.layers.Dense(4, activation='softmax', name='predictions')((x_test,y_test)) +model.fit(x_train, y_train, epochs=10) +outputs = tf.keras.layers.Dense(4, activation='softmax', name='predictions')((x_train,y_train)) model = tf.keras.Model(inputs=inputs, outputs=outputs) model.build() model.save('model')