From c0a46bd9f487f8c54113a5c8bed80e798ea0260b Mon Sep 17 00:00:00 2001 From: Willy Xiao Date: Thu, 5 Mar 2015 10:42:15 -0500 Subject: [PATCH] remove hard-coded output size --- examples/mnist_nnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/mnist_nnet.py b/examples/mnist_nnet.py index 5454f24..3d7038e 100644 --- a/examples/mnist_nnet.py +++ b/examples/mnist_nnet.py @@ -60,8 +60,8 @@ def train(inputs, targets): layer2_dropout, batcher=batcher) # Output layer weights and biases, with random initializations. - W3 = kayak.Parameter( 0.1*npr.randn( layer2_sz, 10 )) - B3 = kayak.Parameter( 0.1*npr.randn(1, 10) ) + W3 = kayak.Parameter( 0.1*npr.randn( layer2_sz, targets.shape[1] )) + B3 = kayak.Parameter( 0.1*npr.randn(1, targets.shape[1]) ) # Output layer. Y = kayak.LogSoftMax( kayak.ElemAdd(kayak.MatMult(H2, W3), B3) )