단순 업데이트
This commit is contained in:
jung-geun
2023-06-28 23:29:27 +09:00
parent 983913f2d2
commit 544a818940
7 changed files with 140884 additions and 67 deletions

View File

@@ -114,6 +114,8 @@ class Particle:
self.best_score = score[1]
self.best_weights = self.model.get_weights()
elif renewal == "loss":
if score[0] == 'nan':
score[0] = np.inf
if score[0] < self.best_score:
self.best_score = score[0]
self.best_weights = self.model.get_weights()