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