【发布时间】:2020-09-03 08:13:22
【问题描述】:
嗨,我是 keras 的新手,我只是想知道 ann 对多项式回归任务有什么好处,或者我们应该只是 以 sklearn 为例,我编写了这个脚本
import numpy as np
import keras
from keras.layers import Dense
from keras.models import Sequential
x=np.arange(1, 100)
y=x**2
model = Sequential()
model.add(Dense(units=200, activation = 'relu',input_dim=1))
model.add(Dense(units=200, activation= 'relu'))
model.add(Dense(units=1))
model.compile(loss='mean_squared_error',optimizer=keras.optimizers.SGD(learning_rate=0.001))
model.fit(x, y,epochs=2000)
但是在对一些数字进行测试后,我没有得到像这样的好结果:
model.predict([300])
array([[3360.9023]], dtype=float32)
我的代码中是否有任何问题,或者我不应该将 ann 用于多项式回归。 谢谢。
【问题讨论】:
标签: python-3.x keras regression