1 changed files with 2 additions and 2 deletions
@@ -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')