papasega commited on
Commit
055dcd6
·
1 Parent(s): 16d1fbe

Update modelutil.py

Browse files
Files changed (1) hide show
  1. modelutil.py +1 -1
modelutil.py CHANGED
@@ -4,7 +4,7 @@ def create_model():
4
 
5
  LAYERS = [tf.keras.layers.Flatten(input_shape=[28,28], name="inputlayer"),
6
  tf.keras.layers.Dense(300, activation='relu', name="hiddenlayer1"),
7
- tf.keras.layers.Dense(128, activation='relu', name="hiddenlayer2"),
8
  tf.keras.layers.Dense(10, activation='softmax', name="outputlayer")]
9
 
10
  model = tf.keras.models.Sequential(LAYERS)
 
4
 
5
  LAYERS = [tf.keras.layers.Flatten(input_shape=[28,28], name="inputlayer"),
6
  tf.keras.layers.Dense(300, activation='relu', name="hiddenlayer1"),
7
+ tf.keras.layers.Dense(100, activation='relu', name="hiddenlayer2"),
8
  tf.keras.layers.Dense(10, activation='softmax', name="outputlayer")]
9
 
10
  model = tf.keras.models.Sequential(LAYERS)