Unity 机器学习代理工具包 (ML-Agents) 是一个开源项目,它使游戏和模拟能够作为训练智能代理的环境。
您最多选择25个主题 主题必须以中文或者字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符
 
 
 
 
 

61 行
2.0 KiB

using System.Collections;
using System.Collections.Generic;
using UnityEngine;
/// CoreBrain which decides actions via communication with an external system such as Python.
public class CoreBrainExternal : ScriptableObject, CoreBrain
{
public Brain brain;
/**< Reference to the brain that uses this CoreBrainExternal */
ExternalCommunicator coord;
/// Creates the reference to the brain
public void SetBrain(Brain b)
{
brain = b;
}
/// Generates the communicator for the Academy if none was present and
/// subscribe to ExternalCommunicator if it was present.
public void InitializeCoreBrain()
{
if (brain.gameObject.transform.parent.gameObject.GetComponent<Academy>().communicator == null)
{
coord = new ExternalCommunicator(brain.gameObject.transform.parent.gameObject.GetComponent<Academy>());
brain.gameObject.transform.parent.gameObject.GetComponent<Academy>().communicator = coord;
coord.SubscribeBrain(brain);
}
else
{
if (brain.gameObject.transform.parent.gameObject.GetComponent<Academy>().communicator is ExternalCommunicator)
{
coord = (ExternalCommunicator)brain.gameObject.transform.parent.gameObject.GetComponent<Academy>().communicator;
coord.SubscribeBrain(brain);
}
}
}
/// Uses the communicator to retrieve the actions, memories and values and
/// sends them to the agents
public void DecideAction()
{
brain.SendActions(coord.GetDecidedAction(brain.gameObject.name));
brain.SendMemories(coord.GetMemories(brain.gameObject.name));
brain.SendValues(coord.GetValues(brain.gameObject.name));
}
/// Uses the communicator to send the states, observations, rewards and
/// dones outside of Unity
public void SendState()
{
coord.giveBrainInfo(brain);
}
/// Nothing needs to appear in the inspector
public void OnInspector()
{
}
}