jung-geun/PSO
1
0
Fork 0
mirror of https://github.com/jung-geun/PSO.git synced 2025-12-19 20:44:39 +09:00
Code Issues Packages Projects Releases Wiki Activity

23-07-11

Browse Source 
mnist one hot 인코딩 적용후 손실 함수 mse 로 변경
This commit is contained in:
jung-geun
2023-07-11 15:00:50 +09:00
parent 026223bd19
commit 2b010c4257
10 changed files with 21 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mnist_tf.py
View File

@@ -62,11 +62,11 @@ 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="categorical_crossentropy", metrics=["accuracy"]
)
# model.compile(
# optimizer="adam", loss="categorical_crossentropy", metrics=["accuracy"]
# )
# model.compile(optimizer="adam", loss="mse", metrics=["accuracy"])
model.compile(optimizer="adam", loss="mse", metrics=["accuracy"])
print("Training model...")
model.fit(x_train, y_train, epochs=100, batch_size=128, verbose=1)