3.5 KiB
Brains
The Brain encapsulates the decision making process. Brain objects must be children of the Academy in the Unity scene hierarchy. Every Agent must be assigned a Brain, but you can use the same Brain with more than one Agent.
Use the Brain class directly, rather than a subclass. Brain behavior is determined by the Brain Type. ML-Agents defines four Brain Types:
- External — The External and Internal types typically work together; set External when training your agents. You can also use the External brain to communicate with a Python script via the Python
UnityEnvironment
class included in the Python portion of the ML-Agents SDK. - Internal – Set Internal to make use of a trained model.
- Heuristic – Set Heuristic to hand-code the agent's logic by extending the Decision class.
- Player – Set Player to map keyboard keys to agent actions, which can be useful to test your agent code.
Each of these types is an implementation of the CoreBrain interface. You can extend the CoreBrain class to create different brain types if the four built-in types don't do what you need.
During training, set your agent's brain type to External. To use the trained model, import the model file into the Unity project and change the brain type to Internal.
The Brain class has several important properties that you can set using the Inspector window. These properties must be appropriate for the agents using the brain. For example, the Vector Observation Space Size
property must match the length of the feature vector created by an agent exactly. See Agents for information about creating agents and setting up a Brain instance correctly.
Brain Properties
The Brain Inspector window in the Unity Editor displays the properties assigned to a Brain component:
Brain Parameters
- Define vector observations, visual observation, and vector actions for the Brain.Vector Observation
Space Type
- Corresponds to whether the observation vector contains a single integer (Discrete) or a series of real-valued floats (Continuous).Space Size
- Length of vector observation for brain (In Continuous space type). Or number of possible values (in Discrete space type).Stacked Vectors
- The number of previous vector observations that will be stacked before being sent to the brain.
Visual Observations
- Describes height, width, and whether to greyscale visual observations for the Brain.Vector Action
Space Type
- Corresponds to whether action vector contains a single integer (Discrete) or a series of real-valued floats (Continuous).Space Size
- Length of action vector for brain (In Continuous state space). Or number of possible values (in Discrete action space).Action Descriptions
- A list of strings used to name the available actions for the Brain.
Type of Brain
- Describes how the Brain will decide actions.External
- Actions are decided by an external process, such as the PPO training process.Internal
- Actions are decided using internal TensorFlowSharp model.Player
- Actions are decided using keyboard input mappings.Heuristic
- Actions are decided using a customDecision
script, which must be attached to the Brain game object.