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-06-22

Browse Source 
np random seed 고정
각 함수의 설명 추가
This commit is contained in:
jung-geun
2023-06-23 04:00:59 +00:00
parent 34729e9b33
commit 953cd44396
8 changed files with 244 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
auto_tunning.py
View File

@@ -14,7 +14,6 @@ from keras import backend as K
# from pso_tf import PSO
from pso import Optimizer
# from optimizer import Optimizer
import numpy as np
@@ -27,7 +26,6 @@ import gc
# print(tf.config.list_physical_devices())
# print(f"Num GPUs Available: {len(tf.config.list_physical_devices('GPU'))}")
def get_data():
(x_train, y_train), (x_test, y_test) = mnist.load_data()