23-07-18

requirements 파일 수정

.gitignore

@@ -12,6 +12,7 @@ pso2keras.egg-info/
 
 # 테스트용 파일
 test.ipynb
+test.py
 
 # 결과 저장용 디렉토리
 result/

conda_env/environment.yaml

@@ -5,12 +5,11 @@ channels:
 dependencies:
   - cudatoolkit=11.2
   - cudnn=8.1.0
-  - matplotlib=3.7.1
   - pandas=1.5.3
   - pip=23.0.1
   - python=3.9.16
   - tqdm=4.65.0
   - pip:
-      - numpy==1.23.5
+      - numpy==1.25.0
       - tensorflow==2.11.0
       - tensorboard==2.11.0

pso/optimizer.py

@@ -234,6 +234,7 @@ class Optimizer:
 
         self.renewal = renewal
         try:
+            train_log_dir = "logs/fit/" + self.day
             if log == 2:
                 assert log_name is not None, "log_name is None"
 
@@ -319,7 +320,6 @@ class Optimizer:
             for epoch in epochs_pbar:
                 max_score = 0
                 min_loss = np.inf
-
                 part_pbar = tqdm(
                     range(len(self.particles)),
                     desc=f"acc : {max_score:.4f} loss : {min_loss:.4f}",
@@ -373,7 +373,15 @@ class Optimizer:
                             del p_
 
                         score = self.particles[i].step_w(
-                            x, y, self.c0, self.c1, w, g_best, w_p, w_g, renewal=renewal
+                            x,
+                            y,
+                            self.c0,
+                            self.c1,
+                            w,
+                            g_best,
+                            w_p,
+                            w_g,
+                            renewal=renewal,
                         )
 
                     else:

requirements.txt

@@ -1,7 +1,6 @@
-ipython @ file:///home/conda/feedstock_root/build_artifacts/ipython_1680185408135/work
+ipython
 keras==2.11.0
-matplotlib @ file:///croot/matplotlib-suite_1679593461707/work
-numpy @ file:///work/mkl/numpy_and_numpy_base_1682953417311/work
+numpy==1.25.0
 pandas==1.5.3
 tensorflow==2.11.1
-tqdm @ file:///croot/tqdm_1679561862951/work
+tqdm==4.65.0