import numpy as np
from sklearn.tree import DecisionTreeRegressor
from sklearn.metrics import classification_report
from sklearn.metrics import mean_squared_error,explained_variance_score
from sklearn.ensemble import AdaBoostRegressor
from sklearn import datasets
from sklearn.utils import shuffle
import matplotlib.pyplot as plt
#导入数据
housing_data = datasets.load_boston()
#打乱数据顺序
X, y = shuffle(housing_data.data, housing_data.target, random_state = 7)
#划分数据集,80%数据用于训练,20%数据用于测试
num_training = int(len(X)*0.8)
X_train, X_test = X[:num_training], X[num_training:]
y_train, y_test = y[:num_training], y[num_training:]
#拟合一个决策树回归模型
dt_regressor = DecisionTreeRegressor(max_depth=4) #选择一个最大深度为4的决策树
dt_regressor.fit(X_train, y_train)
#带AdaBoost算法的决策树回归模型进行拟合
ab_regressor = AdaBoostRegressor(DecisionTreeRegressor(max_depth=4),n_estimators=400,random_state=7)
ab_regressor.fit(X_train, y_train)
#评价决策树回归器训
机器学习-估算房屋价格
最新推荐文章于 2022-07-28 16:19:58 发布