모델과 파티클 클래스에 타입 힌트 추가

This commit is contained in:
jung-geun
2024-03-08 20:15:54 +09:00
parent fa9af45a95
commit 940580b7a6
3 changed files with 35 additions and 28 deletions

View File

@@ -41,7 +41,7 @@ x_train, x_test, y_train, y_test = load_data()
pso_iris = optimizer(
model,
model=model,
loss="categorical_crossentropy",
n_particles=100,
c0=0.5,