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