mirror of
https://github.com/aladdinpersson/Machine-Learning-Collection.git
synced 2026-02-20 13:50:41 +00:00
small revisions to code examples
This commit is contained in:
@@ -130,6 +130,7 @@ for epoch in range(num_epochs):
|
||||
# gradient descent update step/adam step
|
||||
optimizer.step()
|
||||
|
||||
|
||||
# Check accuracy on training & test to see how good our model
|
||||
def check_accuracy(loader, model):
|
||||
num_correct = 0
|
||||
|
||||
@@ -94,6 +94,7 @@ for epoch in range(num_epochs):
|
||||
# gradient descent or adam step
|
||||
optimizer.step()
|
||||
|
||||
|
||||
# Check accuracy on training & test to see how good our model
|
||||
def check_accuracy(loader, model):
|
||||
num_correct = 0
|
||||
|
||||
@@ -95,7 +95,7 @@ for epoch in range(num_epochs):
|
||||
optimizer.step()
|
||||
|
||||
|
||||
# Check accuracy on training & test to see how good our model
|
||||
# Check accuracy on training & test to see how good our model is
|
||||
def check_accuracy(loader, model):
|
||||
num_correct = 0
|
||||
num_samples = 0
|
||||
|
||||
Reference in New Issue
Block a user