1 from sklearn import tree,datasets 2 from sklearn.model_selection import train_test_split 3 wine=datasets.load_wine() 4 X,y=wine.data[:,:2],wine.target 5 X_train,X_test,y_train,y_test=train_test_split(X,y,random_state=3) 6 7 from sklearn.ensemble import RandomForestClassifier 8 forest=RandomForestClassifier(n_estimators=6,random_state=3) 9 forest.fit(X_train,y_train) 10 print("the score of this model:{}".format(forest.score(X_test,y_test)))
1 from matplotlib.colors import ListedColormap 2 cmap_light=ListedColormap(['#FFAAAA','#AAFFAA','#AAAAFF']) 3 cmap_bold=ListedColormap(['#FF0000','#00FF00','#0000FF']) 4 5 import numpy as np 6 from matplotlib import pyplot as plt 7 x_min=X_train[:,0].min()-1 8 x_max=X_train[:,0].max()+1 9 y_min=X_train[:,1].min()-1 10 y_max=X_train[:,1].max()+1 11 xx,yy=np.meshgrid(np.arange(x_min,x_max,.02), 12 np.arange(y_min,y_max,.02)) 13 z=forest.predict(np.c_[xx.ravel(),yy.ravel()]) 14 z=z.reshape(xx.shape) 15 plt.figure() 16 plt.pcolormesh(xx,yy,z,cmap=cmap_light) 17 #plt.pcolormesh(xx,yy,z,cmap=plt.cm.Pastel1) 18 plt.scatter(X[:,0],X[:,1],c=y,cmap=cmap_bold,edgecolors='k',s=20) 19 plt.xlim(xx.min(),xx.max()) 20 plt.ylim(yy.min(),yy.max()) 21 plt.title("Random Forest(n_estimators=6)") 22 plt.show()