您最多选择25个主题
主题必须以中文或者字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符
59 行
1.6 KiB
59 行
1.6 KiB
using System.Collections.Generic;
|
|
using UnityEngine;
|
|
using MLAgents;
|
|
|
|
namespace MLAgents
|
|
{
|
|
|
|
public class RandomDecision : MonoBehaviour, Decision
|
|
{
|
|
BrainParameters brainParameters;
|
|
SpaceType actionSpaceType;
|
|
|
|
public void Awake()
|
|
{
|
|
brainParameters =
|
|
gameObject.GetComponent<Brain>().brainParameters;
|
|
actionSpaceType = brainParameters.vectorActionSpaceType;
|
|
}
|
|
|
|
public float[] Decide(
|
|
List<float> vectorObs,
|
|
List<Texture2D> visualObs,
|
|
float reward,
|
|
bool done,
|
|
List<float> memory)
|
|
{
|
|
if (actionSpaceType == SpaceType.continuous)
|
|
{
|
|
List<float> act = new List<float>();
|
|
|
|
for (int i = 0; i < brainParameters.vectorActionSize[0]; i++)
|
|
{
|
|
act.Add(2 * Random.value - 1);
|
|
}
|
|
|
|
return act.ToArray();
|
|
}
|
|
else
|
|
{
|
|
float[] act = new float[brainParameters.vectorActionSize.Length];
|
|
for (int i = 0; i < brainParameters.vectorActionSize.Length; i++)
|
|
{
|
|
act[i]=Random.Range(0, brainParameters.vectorActionSize[i]);
|
|
}
|
|
return act;
|
|
}
|
|
}
|
|
|
|
public List<float> MakeMemory(
|
|
List<float> vectorObs,
|
|
List<Texture2D> visualObs,
|
|
float reward,
|
|
bool done,
|
|
List<float> memory)
|
|
{
|
|
return new List<float>();
|
|
}
|
|
}
|
|
}
|