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()