machine-learning-deep-learning/Machine-Learning-Collection
1
0
Fork 0
mirror of https://github.com/aladdinpersson/Machine-Learning-Collection.git synced 2026-02-20 13:50:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

small revisions to code examples

Browse Source
This commit is contained in:
Aladdin Persson
2021-03-24 22:07:20 +01:00
parent 31c404822a
commit 80690d56f8
3 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
ML/Pytorch/Basics/pytorch_rnn_gru_lstm.py
View File

@@ -130,7 +130,6 @@ for epoch in range(num_epochs):
# gradient descent update step/adam step # gradient descent update step/adam step
optimizer.step() optimizer.step()
# Check accuracy on training & test to see how good our model # Check accuracy on training & test to see how good our model
def check_accuracy(loader, model): def check_accuracy(loader, model):
num_correct = 0 num_correct = 0

1
ML/Pytorch/Basics/pytorch_simple_CNN.py
View File

@@ -94,7 +94,6 @@ for epoch in range(num_epochs):
# gradient descent or adam step # gradient descent or adam step
optimizer.step() optimizer.step()
# Check accuracy on training & test to see how good our model # Check accuracy on training & test to see how good our model
def check_accuracy(loader, model): def check_accuracy(loader, model):
num_correct = 0 num_correct = 0

2
ML/Pytorch/Basics/pytorch_simple_fullynet.py
View File

@@ -95,7 +95,7 @@ for epoch in range(num_epochs):
optimizer.step() optimizer.step()
# Check accuracy on training & test to see how good our model is # Check accuracy on training & test to see how good our model
def check_accuracy(loader, model): def check_accuracy(loader, model):
num_correct = 0 num_correct = 0
num_samples = 0 num_samples = 0