一个神经元网络

from tensorflow import keras
import numpy as np
#构建模型
model = keras.Sequential([keras.layers.Dense(units=1,input_shape=[1])])
model.compile(optimizer='sgd',loss='mean_squared_error')
#准备训练数据
xs = np.array([-1.0,0.0,1.0,2.0,3.0,4.0],dtype=float)
ys = np.array([-3.0,-1.0,1.0,3.0,5.0,7.0],dtype=float)
#训练模型
model.fit(xs,ys,epochs=500)
#使用模型
print(model.predict([10.0]))

 训练的次数越大,得到的数值越精确

50次

500次

1500次

同时,损耗函数的值越来越小

猜你喜欢

转载自blog.csdn.net/qq_40843903/article/details/114412247