您最多选择25个主题
主题必须以中文或者字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符
25 行
546 B
25 行
546 B
using System.Collections.Generic;
|
|
using Unity.MLAgents;
|
|
using Unity.MLAgents.Extensions.Teams;
|
|
using Unity.MLAgents.Sensors;
|
|
|
|
public class HallwayTeamManager : BaseTeamManager
|
|
{
|
|
List<Agent> m_AgentList = new List<Agent> { };
|
|
|
|
|
|
public override void RegisterAgent(Agent agent)
|
|
{
|
|
m_AgentList.Add(agent);
|
|
}
|
|
|
|
public override void OnAgentDone(Agent agent, Agent.DoneReason doneReason, List<ISensor> sensors)
|
|
{
|
|
agent.SendDoneToTrainer();
|
|
}
|
|
|
|
public override void AddTeamReward(float reward)
|
|
{
|
|
|
|
}
|
|
}
|