使用的数据还是上一篇博客https://blog.csdn.net/HereIcome/article/details/80435395中的数据,只是训练的模型使用random forest regression模型。
from sklearn.ensemble import RandomForestRegressor
regressor=RandomForestRegressor(n_estimators=100,random_state=0) //n_estimators是可以调节的参数。
regressor.fit(x,y) //x,y分别是level、salary
运行结果图:
理解RandomForestRegressor:
参考博客链接:https://blog.csdn.net/qq_16633405/article/details/61200502,链接作者: