Theano学习和门

时间:2016-10-15 13:15:19

标签: python machine-learning neural-network theano

我写了一个简单的神经网络来学习AND门。我试图理解为什么我的成本永远不会降低,预测因子总是0.5:

import numpy as np
import theano
import theano.tensor as T

inputs = [[0,0], [1,1], [0,1], [1,0]]
outputs = [[0], [1], [0], [0]]

x = theano.shared(value=np.asarray(inputs), name='x')
y = theano.shared(value=np.asarray(outputs), name='y')

alpha = 0.1

w_array = np.asarray(np.random.uniform(low=-1, high=1, size=(2, 1)), dtype=theano.config.floatX)
w = theano.shared(value=w_array, name='w', borrow=True)

output = T.nnet.sigmoid(T.dot(x, w))
cost = T.sum((y - output) ** 2)
updates = [(w, w - alpha * T.grad(cost, w))]

train = theano.function(inputs=[], outputs=[], updates=updates)
test = theano.function(inputs=[], outputs=[output])
calc_cost = theano.function(inputs=[], outputs=[cost])

for i in range(60000):
    if (i+1) % 10000 == 0:
        print(i+1)
        print(calc_cost())
    train()    

print(test())

输出始终相同:

10000
[array(1.0)]
20000
[array(1.0)]
30000
[array(1.0)]
40000
[array(1.0)]
50000
[array(1.0)]
60000
[array(1.0)]

[array([[ 0.5],
       [ 0.5],
       [ 0.5],
       [ 0.5]])]

无论输入如何,似乎总是预测0.5,因为在学习期间成本不会偏离1

如果我将输出切换到[[0], [1], [1], [1]]以学习OR门,我会得到正确的预测,并正确降低成本

1 个答案:

答案 0 :(得分:1)

您的模型是

形式
<w, x>

因此它无法建立任何不跨越原点的分离。这样的方程只能表示通过点(0,0)的线,显然,将AND门((1,1)与其他任何东西分开的线)不会越过原点。您必须添加偏见字词,因此您的模型是

<w, x> + b