from sklearn import linear_model
import matplotlib.pyplot as plt
#特征值
X=[[1],[2],[3],[4],[5]]
#目标值
y=[10,20,30,40,50]
#测试值
X_test=[[3.45]]
#线性回归模型
regressor=linear_model.LinearRegression()
#模型训练
regressor.fit(X,y)
#用训练过的模型预测
y_test_pred=regressor.predict(X_test)
print("predition:",X_test,"--->",y_test_pred)
#可视化数据
plt.plot(X,y,linewidth=1,color="green")
plt.scatter(X_test,y_test_pred,c="red",s=100)
plt.show()