Keras实现一个简单的线性回归

这里设置x,y的关系是y=2x+3

因为输入的x是1维的数字，输出的y也是1维的，所以dense层的输出维度为1，总共的参数为（1+1）*1=2个，分别是w和b

# import packages
import numpy as np
from keras.models import *
from keras.layers import *
import matplotlib.pyplot as plt

# generate data
x_train = np.linspace(-1, 1, 200)
y_train = 2 * x_train + np.random.normal(3, 0.5, (x_train.shape[-1]))
x_test = np.linspace(-1, 1, 100)
y_test = 2 * x_test + np.random.normal(3, 0.5, (x_test.shape[-1]))

# initialize model
lr_model = Sequential()
lr_model.add(Dense(1, activation='linear', input_dim=1, name='dense_1'))
lr_model.compile(loss='mse', optimizer='sgd')

# train
lr_model.fit(x_train, y_train, batch_size=32, epochs=500, verbose=1)

# evaluate
x = lr_model.evaluate(x_test, y_test, batch_size=32)
print(lr_model.layers[0].get_weights())
print(lr_model.layer