波士顿房价

from sklearn.datasets import load_boston

boston = load_boston()

print(boston.keys())

print('特征:', boston['feature_names'])

X = boston['data']
Y = boston['target']
print('X shape:', X.shape)
print('Y shape:', Y.shape)

for i in range(2): #查看前两条数据
    print(X[i], Y[i])

from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, Y, test_size=0.2, random_state=1)

from sklearn.linear_model import LinearRegression, Ridge, Lasso
from sklearn.tree import DecisionTreeRegressor
from sklearn.metrics import mean_absolute_error, mean_squared_error, r2_score

#线性回归
lr = LinearRegression()
lr.fit(X_train, y_train)
# print(lr.coef_, lr.intercept_)
y_predict = lr.predict(X_test)
print('线性回归：')
print('mse(均方误差): ', mean_squared_error(y_test, y_predict))
print("mae(平均绝对误差): ", mean_absolute_error(y_test, y_predict))
print("R^2(决定系数): ", lr.score(X_test, y_test))

#岭回归
rr = Ridge()
rr.fit(X_train, y_train)
y_predict = rr.predict(X_test)
print('\n岭回归：')
print('mse(均方误差): ', mean_squared_error(y_test, y_predict))
print("mae(平均绝对误差): ", mean_absolute_error(y_test, y_predict))
print("R^2(决定系数): ", rr.score(X_test, y_test))

#套索回归
lar = Lasso()
lar.fit(X_train, y_train)
y_predict = lar.predict(X_test)
print('\n套索回归：')
print('mse(均方误差): ', mean_squared_error(y_test, y_predict))
print("mae(平均绝对误差): ", mean_absolute_error(y_test, y_predict))
print("R^2(决定系数): ", lar.score(X_test, y_test))

#决策树回归
dtr = DecisionTreeRegressor(max_depth=5)
dtr.fit(X_train, y_train)
y_predict = dtr.predict(X_test)
print('\n回归树：')
print('mse(均方误差): ', mean_squared_error(y_test, y_predict))
print("mae(平均绝对误差): ", mean_absolute_error(y_test, y_predict))
print("R^2(决定系数): ", dtr.score(X_test, y_test))