This time I have made my own environment for ai agent to train. The environment is a 2d game that I modified one of my old game projects. First, create a unity project and add the Unity ML-Agents package into your unity project. Then search for Too Cube Forest in the asset store and download it. I use 2d pixel art cubes from it to build my 2d game. Then, build the ground with cubes. Create a tilemap. Add tilemap collider 2d component to it. Then, export the assets from my old project. And import them to this new project. Then, modify the player script to an agent script. Add CollectObservations, OnActionReceived, OnEpisodeBegin, Heuristic Now let's test the game with heuristic mode. The only thing left before training is the config file. I will need to learn more about it and then make one. The next step is building a config file and then training.