kaggle教程--3--随机森林

随机森林核心代码：

 

1 首先读入excel数据

2 用y存储标签列

3 用X存储所需的特征列

4 train_X, val_X, train_y, val_y = train_test_split(X, y, random_state=1)

下面是随机森林部分

from sklearn.ensemble import RandomForestRegressor
from sklearn.metrics import mean_absolute_error

forest_model = RandomForestRegressor(random_state=1)
forest_model.fit(train_X, train_y)
melb_preds = forest_model.predict(val_X)
print(mean_absolute_error(val_y, melb_preds))