Update PSO and neural network parameters
best score 초기화 를 무작위 값에서 계산 후 설정으로 변경
This commit is contained in:
jung-geun
2023-11-05 17:14:07 +09:00
parent 80695f304d
commit c45ee5873e
7 changed files with 191 additions and 46 deletions

View File

@@ -19,7 +19,7 @@ os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
def make_model():
model = Sequential()
model.add(Dense(12, input_dim=64, activation="relu"))
model.add(Dense(8, activation="relu"))
model.add(Dense(10, activation="relu"))
model.add(Dense(10, activation="softmax"))
return model
@@ -55,7 +55,7 @@ digits_pso = optimizer(
mutation_swarm=0.1,
convergence_reset=True,
convergence_reset_patience=10,
convergence_reset_monitor="acc",
convergence_reset_monitor="loss",
convergence_reset_min_delta=0.001,
)
@@ -66,7 +66,7 @@ digits_pso.fit(
validate_data=(x_test, y_test),
log=2,
save_info=True,
renewal="acc",
renewal="loss",
log_name="digits",
)