Improving model for playing Tetris

I’m using TensorFlow.js / Keras to train a model that would automatically solve Tetris in the browser.

Currently the model is not very good; what would be the next steps in learning how to improve it? – I’m learning both the theory and the TensorFlow / Keras APIs at the same time, so it is a little difficult.

My current idea is that I somehow need to introduce memory into the model, or parametrise the reward function; not sure how to do neither theoretically nor in practically.

I am not very familiar either with this kind of problem but maybe you could have a look on Reinforcement Learning, it might be suitable for it.