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

Merge branch 'neural_network' of D:\Documents\Studia\Semestr-4\sztuczna-inteligencja\super-saper with conflicts.

Browse Source
This commit is contained in:
dar-gol 2021-05-30 15:46:45 +02:00
parent 72c6c5e2d8
commit 61419f5f64
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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]