jung-geun/PSO
1
0
Fork 0
mirror of https://github.com/jung-geun/PSO.git synced 2025-12-20 04:50:45 +09:00
Code Issues Packages Projects Releases Wiki Activity

23-08-06

Browse Source 
메모리 누수 다소 해결
Fixes #2
EBPSO 의 구현 부분의 문제가 있어 수정중
This commit is contained in:
jung-geun
2023-08-06 19:14:44 +09:00
parent f3b61e280f
commit 8d558d0f26
12 changed files with 201 additions and 161 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
iris.py
View File

@@ -11,7 +11,7 @@ from tensorflow import keras
from tensorflow.keras import layers
from tensorflow.keras.models import Sequential
from pso import Optimizer
from pso import optimizer
def make_model():
@@ -42,7 +42,7 @@ x_train, x_test, y_train, y_test = load_data()
loss = ["categorical_crossentropy", "mean_squared_error"]
pso_iris = Optimizer(
pso_iris = optimizer(
model,
loss=loss[1],
n_particles=100,
@@ -63,7 +63,7 @@ best_score = pso_iris.fit(
save_info=True,
log=2,
log_name="iris",
save_path="./result/iris",
save_path="result/iris",
renewal="acc",
check_point=25,
)