sklearn机器学习之回归决策树以及参数设置(波士顿房价数据集)

1.导入相应包以及波士顿房价数据集

from sklearn.datasets import load_boston
from sklearn.model_selection import cross_val_score
from sklearn.tree import DecisionTreeRegressor

2.建立模型并进行验证

这里采用sklearn内置函数进行十折交叉验证,打分函数使用负均方误差

boston = load_boston()
#生成回归器
regressor = DecisionTreeRegressor(random_state=0)
#交叉验证
cross_val_score(regressor, boston.data, boston.target, cv=10, scoring="neg_mean_squared_error")

你可能感兴趣的:(决策树,python,机器学习,人工智能)