File size: 1,753 Bytes
05c9ac2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
using System;
using Unity.MLAgents.Sensors;
namespace Unity.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[] CreateSensors()
{
return new ISensor[] { new BasicSensor(basicController) };
}
}
/// <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.position] = 1;
}
/// <inheritdoc/>
public override ObservationSpec GetObservationSpec()
{
return ObservationSpec.Vector(BasicController.k_Extents);
}
/// <inheritdoc/>
public override string GetName()
{
return "Basic";
}
}
}
|