模型无法学习-keras

时间:2018-11-06 18:11:02

标签: machine-learning neural-network keras

背景

我有一个非常简单的脚本,它创建了一个keras模型,旨在像XOR门一样工作。

我在get_data函数中生成了40000个数据点。它创建两个数组;一个输入数组,按顺序包含1和0,输出为1或0。

问题

当我运行代码时,似乎并没有学习到,每次训练后得到的结果都有很大的不同。

代码

from keras import models
from keras import layers

import numpy as np

from random import randint


def get_output(a, b): return 0 if a == b else 1


def get_data ():
    data = []
    targets = []

    for _ in range(40010):
        a, b = randint(0, 1), randint(0, 1)

        targets.append(get_output(a, b))
        data.append([a, b])

    return data, targets


data, targets = get_data()

data = np.array(data).astype("float32")
targets = np.array(targets).astype("float32")

test_x = data[40000:]
test_y = targets[40000:]

train_x = data[:40000]
train_y = targets[:40000]

model = models.Sequential()

# input
model.add(layers.Dense(2, activation='relu', input_shape=(2,)))

# hidden
# model.add(layers.Dropout(0.3, noise_shape=None, seed=None))
model.add(layers.Dense(2, activation='relu'))
# model.add(layers.Dropout(0.2, noise_shape=None, seed=None))
model.add(layers.Dense(2, activation='relu'))

# output
model.add(layers.Dense(1, activation='sigmoid')) # sigmoid puts between 0 and 1

model.summary() # print out summary of model

model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])

res = model.fit(train_x, train_y, epochs=2000, batch_size=200, validation_data=(test_x, test_y)) # train

print 'predict: \n', test_x
print model.predict(test_x)

输出

[[0. 1.]
 [1. 1.]
 [1. 1.]
 [0. 0.]
 [1. 0.]
 [0. 0.]
 [0. 0.]
 [0. 1.]
 [1. 1.]
 [1. 0.]]
[[0.6629775 ]
 [0.00603844]
 [0.00603844]
 [0.6629775 ]
 [0.6629775 ]
 [0.6629775 ]
 [0.6629775 ]
 [0.6629775 ]
 [0.00603844]
 [0.6629775 ]]

编辑

即使没有辍学层,我也得到了非常相似的结果。

1 个答案:

答案 0 :(得分:3)

您的问题有几个问题。

首先,您的导入是非正统的(与您的问题无关,是的,但这确实有助于遵守某些约定):

from keras.models import Sequential
from keras.layers import Dense
import numpy as np

第二,关于XOR问题,您不需要成千上万的示例。只有四种组合:

X = np.array([[0,0],[0,1],[1,0],[1,1]])
y = np.array([[0],[1],[1],[0]])

仅此而已。

第三,出于同样的原因,您实际上无法使用XOR获得“验证”或“测试”数据;在最简单的方法(即您可以尝试在此处尝试做的事情)中,您只能使用这4种组合来测试模型对函数的学习程度(因为没有更多!)。

第四,您应该从一个简单的单层模型(具有2个以上的单元,并且没有丢失)开始,然后在需要时逐步进行

model = Sequential()
model.add(Dense(8, activation="relu", input_dim=2))
model.add(Dense(1, activation="sigmoid"))

model.compile(loss='binary_crossentropy', optimizer='adam')
model.fit(X, y, batch_size=1, epochs=1000)

这应该使您的损失降低至〜0.12;它对该功能的了解程度如何?

model.predict(X)
# result:
array([[0.31054294],
       [0.9702552 ],
       [0.93392825],
       [0.04611744]], dtype=float32)

y
# result:
array([[0],
       [1],
       [1],
       [0]])

这样够好吗?好吧,我不知道-正确的答案永远是“取决于”!但是您现在有了一个起点(可以说是一个学习的网络),您可以从中进行进一步的实验...