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-07-11

Browse Source 
one hot 인코딩 적용
This commit is contained in:
jung-geun
2023-07-11 02:10:14 +09:00
parent 0dd6cc4cab
commit 026223bd19
4 changed files with 16 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
mnist_tf.py
View File

@@ -59,8 +59,11 @@ def make_model():
model = make_model()
x_train, y_train, x_test, y_test = get_data()
y_train = tf.one_hot(y_train, 10)
y_test = tf.one_hot(y_test, 10)
model.compile(
optimizer="adam", loss="sparse_categorical_crossentropy", metrics=["accuracy"]
optimizer="adam", loss="categorical_crossentropy", metrics=["accuracy"]
)
# model.compile(optimizer="adam", loss="mse", metrics=["accuracy"])