Fig 5 - Code segment for creating the Keras model