代码如下:
from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import load_wine
from sklearn.model_selection import train_test_split
wine = load_wine()
data = wine.data
target = wine.target
traindata,testdata,traintarget,testtarget = train_test_split(data,target,test_size=0.2,random_state=1234)
model_rf = RandomForestClassifier()
model_rf.fit(traindata,traintarget)
testtarget_pre = model_rf.predict(testdata)
print("预测:",testtarget_pre[:20])
print("实际:",testtarget[:20])
from sklearn.metrics import accuracy_score,confusion_matrix
print("预测准确率:",accuracy_score(testtarget,testtarget_pre))
print("实际准确率:",accuracy_score(testtarget,testtarget_pre))
结果如下: