s452686/Super-Saper
s452686/Super-Saper
4
5
Fork 0
Code Issues Pull Requests Releases Wiki Activity

neural_network #4

Merged
s452622 merged 34 commits from neural_network into master 2021-06-08 23:47:22 +02:00
Conversation 0 Commits 34 Files Changed 879 +6 -7
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 61419f5f64 - Show all commits

3
src/machine_learning/neural_network/learning_neural_network.py
View File

@ -30,7 +30,6 @@ if __name__ == '__main__':
for epoch in range(n_iter): for epoch in range(n_iter):
for image, label in zip(train_images, train_labels): for image, label in zip(train_images, train_labels):
print(image.shape)
optimizer.zero_grad() optimizer.zero_grad()
output = model(image) output = model(image)
@ -44,7 +43,7 @@ if __name__ == '__main__':
train(model, 100) train(model, 100)
# def accuracy(expected, predicted): # def accuracy(expected, predicted):
# return len([1 for e, p in zip(expected, predicted) if e == p]) / len(expected) # return len([_ for e, p in zip(expected, predicted) if e == p]) / len(expected)
# #
# #
# predicted = [model(image).argmax() for image in train_images] # predicted = [model(image).argmax() for image in train_images]