Guest User

Untitled

a guest
Jan 21st, 2019
70
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.50 KB | None | 0 0
  1. def build_model():
  2. model = keras.Sequential([
  3. layers.Dense(32, activation=tf.nn.sigmoid, input_shape=[len(trainingData.keys())]),
  4. layers.Dense(32, activation=tf.nn.sigmoid),
  5. layers.Dense(32, activation=tf.nn.sigmoid),
  6. layers.Dense(32, activation=tf.nn.sigmoid),
  7. layers.Dense(1)
  8. ])
  9.  
  10. optimizer = tf.train.RMSPropOptimizer(0.01)
  11.  
  12. model.compile(loss='mse',
  13. optimizer=optimizer,
  14. metrics=['mae', 'mse'])
  15. return model
Add Comment
Please, Sign In to add comment