s452701/AI-Project
3
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Plot the mean scores

Browse Source
This commit is contained in:
Kanewersa 2021-06-07 13:39:11 +02:00
parent 9f86f7dd93
commit 88f13d7d0d
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
survival/learning_utils.py
View File

@ -14,6 +14,7 @@ class LearningUtils:
self.plot_mean_scores = []
self.total_score = 0
self.last_actions: [Action, [int, int]] = []
self.plots = 0
def add_scores(self, learning: LearningComponent, games_count: int):
self.plot_scores.append(learning.score)
@ -25,14 +26,16 @@ class LearningUtils:
display.clear_output(wait=True)
display.display(plt.gcf())
plt.clf()
plt.title('Training...')
plt.title('Results')
plt.xlabel('Number of Games')
plt.ylabel('Score')
plt.plot(self.plot_scores)
# plt.plot(self.plot_mean_scores)
plt.plot(self.plot_mean_scores)
plt.ylim(ymin=0)
plt.text(len(self.plot_scores) - 1, self.plot_scores[-1], str(self.plot_scores[-1]))
# plt.text(len(self.plot_mean_scores) - 1, self.plot_mean_scores[-1], str(self.plot_mean_scores[-1]))
plt.text(len(self.plot_mean_scores) - 1, self.plot_mean_scores[-1], str(self.plot_mean_scores[-1]))
self.plots += 1
plt.savefig(f'model/plots/{self.plots}.png')
plt.show(block=False)
plt.pause(.1)