使用迴歸樹進行波士頓房價的預測並且可視化樹的結構

本文主要實現使用迴歸決策樹對波士頓房價的預測

#使用boston房價預測練習迴歸樹的建模
import pandas as pd
from sklearn import preprocessing
from sklearn import tree
from sklearn.datasets import load_boston
boston_house=load_boston()
#加載數據
boston_features_name=boston_house.feature_names
boston_features=boston_house.data
boston_taget=boston_house.target
#構建模型
rgs=tree.DecisionTreeRegressor(max_depth=4)
rgs=rgs.fit(boston_features,boston_taget)
print(rgs)
import pydotplus
from IPython.display import Image,display
dot_data=tree.export_graphviz(rgs,
               out_file=None,
               feature_names=boston_features_name,
               class_names=boston_taget,
               filled=True,
               rounded=True
               )
graph=pydotplus.graph_from_dot_data(dot_data)
display(Image(graph.create_png()))

標題

標題