keras 非线性回归:为什么模型不学习?

vohkndzv  于 2022-11-24  发布在  其他
关注(0)|答案(1)|浏览(195)

我刚开始学习keras。我试图在keras中训练一个非线性回归模型,但模型似乎没有学到很多。

#datapoints
X = np.arange(0.0, 5.0, 0.1, dtype='float32').reshape(-1,1)
y = 5 * np.power(X,2) + np.power(np.random.randn(50).reshape(-1,1),3)

#model
model = Sequential()
model.add(Dense(50, activation='relu', input_dim=1))
model.add(Dense(30, activation='relu', init='uniform'))
model.add(Dense(output_dim=1, activation='linear'))

#training
sgd = SGD(lr=0.1);
model.compile(loss='mse', optimizer=sgd, metrics=['accuracy'])
model.fit(X, y, nb_epoch=1000)

#predictions
predictions = model.predict(X)

#plot
plt.scatter(X, y,edgecolors='g')
plt.plot(X, predictions,'r')
plt.legend([ 'Predictated Y' ,'Actual Y'])
plt.show()

我做错了什么?

8wtpewkr

8wtpewkr1#

你的学习速度太快了。
此外,与您的问题无关,但您不应要求metrics=['accuracy'],因为这是一个回归设置和accuracy is meaningless
因此,通过这些更改:

sgd = SGD(lr=0.001);
model.compile(loss='mse', optimizer=sgd)

plt.legend([ 'Predicted Y' ,'Actual Y']) # typo in legend :)

以下是一些输出(由于y中的随机元素,不同轮次的结果会有所不同):

相关问题