波士顿房价数据集——随机森林

波士顿房价数据链接：https://pan.baidu.com/s/1JPrcNl1AgNCKEHCjOGyHvQ 
提取码：1234

import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split
from sklearn import metrics  #评价函数库
import pandas as pd
import numpy as np
from sklearn.ensemble import RandomForestRegressor #导入随机森林
from sklearn.model_selection import GridSearchCV  #网格搜索验证
from sklearn import tree 
import pydotplus #绘制随机森林
from IPython.display import Image,display #显示图像
%matplotlib inline  #在当前环境中显示图像

df = pd.read_csv("D:/波士顿房价预测/boston_housing_data.csv")
df.dropna(inplace=True)  #消除空值
x = df.drop(["MEDV"],axis = 1)  #x选取前13个特征
y = df["MEDV"]  #y选取房价
x_train,x_test,y_train,y_test = train_test_split(x,y,random_state = 0)

#定义网格搜索
param_grid = {
    "n_estimators":[5,10,20,100,200],   #数值均可预设
    "max_depth":[3,5,7],
    "max_features":[0.6,0.7,0.8,1]
}