def onEnd(self, agent):
if (self._active == False):
return
bestIndex = np.argmax(self._validationScores)
print("Best neural net obtained after {} epochs, with validation score {}".format(bestIndex+1, self._validationScores[bestIndex]))
if self._testID != None:
print("Test score of this neural net: {}".format(self._testScores[bestIndex]))
try:
os.mkdir("scores")
except Exception:
pass
basename = "scores/" + self._filename
joblib.dump({"vs": self._validationScores, "ts": self._testScores}, basename + "_scores.jldump")
评论列表
文章目录