keras模拟线性回归

最新推荐文章于 2023-09-12 12:42:25 发布

zjLOVEcyj

最新推荐文章于 2023-09-12 12:42:25 发布

阅读量89

点赞数

分类专栏： keras 文章标签：深度学习

本文链接：https://blog.csdn.net/cyj5201314/article/details/118281721

版权

keras 专栏收录该内容

10 篇文章 0 订阅

订阅专栏

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

X = np.linspace(-1, 1, 100)
noise = np.random.normal(0, 0.01, X.shape)
Y = X * 2 + 3 + noise

model = Sequential()
model.add(Dense(1, input_dim=1))

model.compile(optimizer="sgd", loss="mse")

for i in range(50):
    model.fit(X, Y, batch_size=20)
    W, b = model.layers[0].get_weights()
    print("W=", W, "b=", b)