Update m1_tf_test.py

2022-07-10 22:53:56 +02:00
parent ac0ec7211b
commit 8e8e0e08a2
1 changed files with 1 additions and 1 deletions
@@ -33,7 +33,7 @@ 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)
+outputs = tf.keras.layers.Dense(4, activation='softmax', name='predictions')((x_test,y_test))
 model = tf.keras.Model(inputs=inputs, outputs=outputs)
 model.build()
 model.save('model')