使用线性回归模型在比萨训练样本上进行拟合

训练数据和测试数据

代码：

x_train = [[6], [8], [10], [14], [18]]
y_train = [[7], [9], [13], [17.5], [18]]

from sklearn.linear_model import LinearRegression
regressor = LinearRegression()
regressor.fit(x_train, y_train)

import numpy as np
xx = np.linspace(0, 26, 100)#生成了1行100列的一个矩阵
#print(xx)
xx = xx.reshape(xx.shape[0], 1)#将1行100列的矩阵转化成100行1列的矩阵形式
#print(xx)

yy = regressor.predict(xx)

import matplotlib.pyplot as plt
plt.scatter(x_train, y_train)
plt1, = plt.plot(xx, yy, label="Degree=1")

plt.axis([0, 25, 0, 25])
plt.xlabel('Diameter of Pizza')
plt.ylabel('Price of Pizza')
plt.legend(handles=[plt1])
plt.show()

print('The R-squared value of Linear Regressor performing on the training data is', regressor.score(x_train, y_train))

运行结果如下：

The R-squared value of Linear Regressor performing on the training data is 0.910001596424

效果图如下：