23-07-17

tensorboard 적용된 pypi
log 저장 디렉토리 지정해야 tensorboard 사용 가능
pypi 소스파일 제거
tqdm 자동으로 변경

iris.py

@@ -40,7 +40,7 @@ def load_data():
 model = make_model()
 x_train, x_test, y_train, y_test = load_data()
 
-loss = ["categorical_crossentropy", 'mean_squared_error']
+loss = ["categorical_crossentropy", "mean_squared_error"]
 
 pso_iris = Optimizer(
     model,
@@ -60,11 +60,11 @@ best_score = pso_iris.fit(
     x_train,
     y_train,
     epochs=200,
-    save=True,
+    save_info=True,
+    log=2,
+    log_name="iris",
     save_path="./result/iris",
     renewal="acc",
-    empirical_balance=False,
-    Dispersion=False,
     check_point=25,
 )