-
Notifications
You must be signed in to change notification settings - Fork 0
/
ExtraPoint2.py
23 lines (20 loc) · 1.12 KB
/
ExtraPoint2.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# Extra point 2
import numpy as np
import matplotlib.pyplot as plt
from part5 import reading_files, start_learning_with_vectorization, calculate_accuracy
# initialize parameters
n_h1 = n_h2 = 16 # hidden layer 1, 2
n_x = 784 # size of the input layer
n_y = 10 # size of the output layer
learning_rate, number_of_epochs, batch_size, number_of_images = 1, 5, 50, 60000
if __name__ == "__main__":
train_set, test_set = reading_files()
W1, W2, W3, b1, b2, b3, total_cost_arr_in_batch = start_learning_with_vectorization(train_set, number_of_epochs,
batch_size, learning_rate,
"tanh") # this W,b after learning
accuracy_in_train_set = calculate_accuracy(W1, W2, W3, b1, b2, b3, train_set, "tanh")
print("Accuracy in train set :", accuracy_in_train_set)
accuracy_in_test_set = calculate_accuracy(W1, W2, W3, b1, b2, b3, test_set, "tanh")
print("Accuracy in test set :", accuracy_in_test_set)
plt.plot(total_cost_arr_in_batch)
plt.show()