1. 程式人生 > >決策樹的python程式碼實現

決策樹的python程式碼實現

關於什麼是決策樹這種可以用百度解決的問題就不在這裡貼出來了,這裡只講程式碼實現。
from sklearn.datasets import load_iris
iris=load_iris()
# print(iris)
# print(len(iris["data"]))#150個數據
from sklearn.model_selection import train_test_split
#把資料分為測試資料和驗證資料 test_size=0.2驗證資料集佔20% 也就是150個數據有30個驗證集 random_state=1隨機的選擇30個數據
train_data,test_data,train_target,test_target=train_test_split(iris.data,iris.target,test_size=0.2,random_state=1)