We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 87cf988 commit f29b8f5Copy full SHA for f29b8f5
examples/mnist_small.py
@@ -30,7 +30,7 @@ def test(net, test_data):
30
print(i)
31
32
t = test_row[0]
33
- x = to_col(test_row[1:])
+ x = to_col(test_row[1:])/255
34
out = net.forward_pass(x)
35
guess = np.argmax(out)
36
if t == guess:
@@ -43,7 +43,7 @@ def train(net, train_data):
43
if not i%1000:
44
45
46
- net.train(to_col(train_row[1:]), train_row[0])
+ net.train(to_col(train_row[1:])/255, train_row[0])
47
48
49
if __name__ == "__main__":
0 commit comments