Update m1_tf_test.py

This commit is contained in:
Smiril
2022-07-11 10:44:31 +02:00
committed by GitHub
parent 788d1e57a2
commit a02cfd461e

View File

@@ -89,7 +89,6 @@ model = tf.keras.models.Sequential([
tf.keras.layers.Dropout(0.2)
])
loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True)
model.add(LSTM(1, input_shape=(10, 1)))
model.compile(optimizer='adam', loss=loss_fn, metrics=['accuracy'])
these = model.fit(x_train, y_train, epochs=10).history
that = ComputeSum(len(these))