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

69 行
1.8 KiB

using System;
using MLAgents.Sensors;
using UnityEngine.Serialization;
namespace MLAgentsExamples
{
/// <summary>
/// A simple example of a SensorComponent.
/// This should be added to the same GameObject as the BasicController
/// </summary>
public class BasicSensorComponent : SensorComponent
{
public BasicController basicController;
/// <summary>
/// Creates a BasicSensor.
/// </summary>
/// <returns></returns>
public override ISensor CreateSensor()
{
return new BasicSensor(basicController);
}
/// <inheritdoc/>
public override int[] GetObservationShape()
{
return new[] { BasicController.k_Extents };
}
}
/// <summary>
/// Simple Sensor implementation that uses a one-hot encoding of the Agent's
/// position as the observation.
/// </summary>
public class BasicSensor : SensorBase
{
public BasicController basicController;
public BasicSensor(BasicController controller)
{
basicController = controller;
}
/// <summary>
/// Generate the observations for the sensor.
/// In this case, the observations are all 0 except for a 1 at the position of the agent.
/// </summary>
/// <param name="output"></param>
public override void WriteObservation(float[] output)
{
// One-hot encoding of the position
Array.Clear(output, 0, output.Length);
output[basicController.m_Position] = 1;
}
/// <inheritdoc/>
public override int[] GetObservationShape()
{
return new[] { BasicController.k_Extents };
}
/// <inheritdoc/>
public override string GetName()
{
return "Basic";
}
}
}