// Start is called before the first frame update
void Start()
{
trainingRewards = new float [totalEpisodes];
trainedRewards = new float [totalEpisodes];
menuManagerScript = menuManager.GetComponent <MenuManagerScript>();
menuManagerScript.UpdateEpisodeText(episodeCount, totalEpisodes);
qTableScript = QTable.GetComponent <QTableScript>();
}
public void ResetSimulator()
{
KillAllCans();
firstTime = 1;
MakeCans();
PlaceRobby();
//Switch from training to trained
if (episodeCount >= totalEpisodes + 1)
{
if (training == true)
{
episodeCount = 1;
training = false;
}
else
{
allDone = true;
}
}
if (training == true)
{
trainingRewardchart.DataSource.AddPointToCategory("Thingy", episodeCount, rewardThisEpisode);
trainingRewards[episodeCount] = (float)rewardThisEpisode;
totalTrainingReward += rewardThisEpisode;
UpdateTrainingVals();
}
else
{
trainedModelRewardChart.DataSource.AddPointToCategory("Thingy", episodeCount, rewardThisEpisode);
trainedRewards[episodeCount] = (float)rewardThisEpisode;
totalTrainedReward += rewardThisEpisode;
UpdateTrainedVals();
}
rewardThisEpisode = 0;
episodeCount++;
menuManagerScript.UpdateEpisodeText(episodeCount, totalEpisodes);
}