import numpy as np
from math import sqrt
import matplotlib as mpl
import matplotlib.pyplot as plt
from sklearn import datasets
from sklearn.metrics import r2_score
from sklearn.metrics import mean_absolute_error
from sklearn.metrics import mean_squared_error
from sklearn.model_selection import train_test_split#将数据分为测试集和训练集
from sklearn.linear_model import LinearRegression
boston=datasets.load_boston()
print(boston.feature_names)
x=boston.data
y=boston.target
x=x[y<50]
y=y[y<50]
X_train,X_test,y_train,y_test=train_test_split(x,y,test_size=0.2,random_state=666)
reg1 = LinearRegression()
reg1.fit(X_train,y_train)
print(reg1.coef_)
print(reg1.intercept_)
print(reg1.score(X_test, y_test))
#
# y_predict=reg1.predict(X_test)
#
# plt.scatter(X_train,y_train)
# plt.plot(X_train,reg1.predict(X_train),color="r")
# plt.show()