batuhan-basoglu/Classification-of-Image-Data-with-MLP-and-CNN
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added the experiment 1.

Browse source
This commit is contained in:
Batuhan Berk Başoğlu 2025-11-16 19:23:34 -05:00
parent c2e8cfa06d
commit 901f472da1
Signed by: batuhan-basoglu
SSH key fingerprint: SHA256:kEsnuHX+qbwhxSAXPUQ4ox535wFHu/hIRaa53FzxRpo
9 changed files with 254 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
multilayer-perceptron.py
View file

@ -154,11 +154,11 @@ y_train = train_set.targets.numpy()
x_test = test_set.data.numpy().reshape(-1, 28 * 28).astype(np.float32) / 255.0
y_test = test_set.targets.numpy()
# MLP Initialization
# MLP initialization
mlp = MLP(
input_size=28 * 28,
hidden_size1=128,
hidden_size2=64,
hidden_size1=256,
hidden_size2=256,
output_size=10,
weight_scale=1e-2
)
@ -171,7 +171,7 @@ mlp.fit(
y_val=y_test,
lr=1e-2,
epochs=10,
batch_size=128
batch_size=256
)
# tests the model