Experimenting with Unity Machine Learning – Making a New Learning Environment

Reading Time: 1 minute

I’ve created a new learning environment. The tutorial is great, but for me, most interesting part was to train the agent myself.

Unfortunately Testing the Environment explains how to test it in Unity play mode using Player Brain. I want more. So I’ve tried to train External Brain for my own. And I’ve made it!

I renamed Brain in Hierarchy to RollerBrain

RollerBrain

Then I copied  Unity-Technologies/ml-agents \python\trainer_config.yaml SmallWallBrain and create RollerBrain with parameters

RollerBrain yaml

Here, you can get working trainer_config.yaml file.

And voilà, as you can see in the video, the agent works perfectly.

Now, it’s time to understand more deeply how is this all working! ;p

Leave a Reply

This site uses Akismet to reduce spam. Learn how your comment data is processed.