On running the code as it is on the provided data I get the following error :

python: 3.7
tf: 1.15
keras: 2.3.1

InvalidArgumentError: Incompatible shapes: [21504] vs. [1024,21]
[[{{node metrics_11/acc/Equal}}]]