Dataset for the spaceship task from "Metacontrol for Adaptive Imagination-Based Optimization" - View it on GitHub