This gets rid of the UserWarning: `The 'lr' argument is deprecated, use 'learning_rate' instead.`
This fixes the error: `Input 0 of layer "dense" is incompatible with the layer: expected min_ndim=2, found ndim=1. Full shape received: (5,)`, which currently makes the example codes not run.