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

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