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

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>();
}
}
}