Update m1_tf_test.py

2022-07-10 22:58:22 +02:00
parent 8e8e0e08a2
commit 12eaf87335
1 changed files with 2 additions and 2 deletions
--- 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')