Update m1_tf_test.py

2022-07-11 10:42:04 +02:00
parent dd07ea8ff3
commit 788d1e57a2
1 changed files with 1 additions and 1 deletions
@@ -89,7 +89,7 @@ model = tf.keras.models.Sequential([
 tf.keras.layers.Dropout(0.2)
 ])
 loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True)
-model.add(LSTM(28, input_shape=(10, 1)))
+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))