大家好,我是菜鸟君,之前跟大家聊过R语言的随机森林建模,指路 R语言 | 随机森林建模实战(代码+详解),作为刚过完1024节日的码农算法工程师来说,怎么可能只会用一种语言呢?今天就来说说Python怎么进行随机森林的模型构建。
首先,加载一些我们需要的库。需要注意的是,我们想进行分类预测,所以加载的是随机森林分类功能。如果想进行回归预测,需要加载随机森林回归功能哦(RandomForestRegression)。
# -*- coding: utf-8 -*-from sklearn.datasets import load_iris #iris数据集from sklearn.ensemble import RandomForestClassifier #随机森林分类from sklearn import metrics #模型结果指标库import pandas as pd import numpy as npimport matplotlib.pyplot as plt #画图
然后,我们加载数据集。今天用到的是自带的鸢尾花数据集。
# 还是用鸢尾花数据集吧iris=load_iris()#iris的4个属性,标签是花的种类print (iris['target'].s