File size: 1,174 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 |
using System;
using System.Collections.Generic;
using UnityEngine;
using Unity.MLAgents.Policies;
namespace Unity.MLAgents.Demonstrations
{
/// <summary>
/// Summary of a loaded Demonstration file. Only used for display in the Inspector.
/// </summary>
[Serializable]
internal class DemonstrationSummary : ScriptableObject
{
public DemonstrationMetaData metaData;
public BrainParameters brainParameters;
public List<ObservationSummary> observationSummaries;
public void Initialize(BrainParameters brainParams,
DemonstrationMetaData demonstrationMetaData, List<ObservationSummary> obsSummaries)
{
brainParameters = brainParams;
metaData = demonstrationMetaData;
observationSummaries = obsSummaries;
}
}
/// <summary>
/// Summary of a loaded Observation. Currently only contains the shape of the Observation.
/// </summary>
/// <remarks>This is necessary because serialization doesn't support nested containers or arrays.</remarks>
[Serializable]
internal struct ObservationSummary
{
public int[] shape;
}
}
|