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-10-18

Browse Source 
모델 save 시 score 확인 추가
배치 사이즈 오류 수정
This commit is contained in:
jung-geun
2023-10-18 16:40:52 +09:00
parent eb8daffdea
commit 6c6aa221f8
4 changed files with 76 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
fashion_mnist.py
View File

@@ -33,23 +33,6 @@ def get_data():
return x_train, y_train, x_test, y_test
def get_data_test():
(x_train, y_train), (x_test, y_test) = fashion_mnist.load_data()
x_test = x_test / 255.0
x_test = x_test.reshape((10000, 28, 28, 1))
y_train, y_test = tf.one_hot(y_train, 10), tf.one_hot(y_test, 10)
x_train, x_test = tf.convert_to_tensor(
x_train), tf.convert_to_tensor(x_test)
y_train, y_test = tf.convert_to_tensor(
y_train), tf.convert_to_tensor(y_test)
print(f"x_test : {x_test[0].shape} | y_test : {y_test[0].shape}")
return x_test, y_test
def make_model():
model = Sequential()
model.add(