The challenge is using all 3 Genesis games.
From the downloaded CSV of game states, all stages from the 3 games are valid training stages, and the custom stages used for testing are derived from all 3 games. Yikes.
I really hope those custom stages use the same art assets as the original games.
For an example of a system that does learn new rules using a single model check out this post from Vicarious[0].
[0] https://www.vicarious.com/2017/08/07/general-game-playing-wi...