紅酒數(shù)據(jù)決策樹分類

數(shù)據(jù)挖掘技術(shù)與應(yīng)用紅酒數(shù)據(jù)決策樹分類實(shí)訓(xùn)實(shí)訓(xùn)目標(biāo)理解決策樹算法的核心步驟。利用Python實(shí)現(xiàn)算法應(yīng)用，提升編程技能。通過(guò)數(shù)據(jù)集的部分樣本訓(xùn)練構(gòu)造決策樹模型。調(diào)用構(gòu)建好的決策樹模型對(duì)測(cè)試集樣本進(jìn)行預(yù)測(cè)，并求出測(cè)試精度。實(shí)訓(xùn)環(huán)境使用3.6版本的Python。使用jupyternotebook或PyCharm2018社區(qū)版作為代碼編輯器。numpy、pandas、sklearn實(shí)訓(xùn)數(shù)據(jù)紅酒數(shù)據(jù)共178個(gè)樣本，代表了紅酒的三個(gè)檔次（分別有59，71，48個(gè)樣本），以及與之對(duì)應(yīng)的13維的屬性數(shù)。實(shí)訓(xùn)內(nèi)容導(dǎo)入必要的庫(kù)。導(dǎo)入數(shù)據(jù)集。獲取特征值。獲取標(biāo)簽。隨機(jī)劃分訓(xùn)練集和測(cè)試集創(chuàng)建模型對(duì)測(cè)試集進(jìn)行預(yù)測(cè)并將結(jié)果與label進(jìn)行對(duì)比，計(jì)算準(zhǔn)確度紅酒數(shù)據(jù)決策樹分類實(shí)現(xiàn)代碼導(dǎo)入數(shù)據(jù)包fromsklearn.model_selectionimporttrain_test_splitfromsklearn.treeimportDecisionTreeClassifierfromsklearnimportdatasetsimportnumpyasnp數(shù)據(jù)說(shuō)明紅酒數(shù)據(jù)共178個(gè)樣本，代表了紅酒的三個(gè)檔次（分別有59，71，48個(gè)樣本），以及與之對(duì)應(yīng)的13維的屬性數(shù)。導(dǎo)入數(shù)據(jù)集wine=datasets.load_wine()獲取特征值X=wine.data獲取標(biāo)簽Y=wine.target隨機(jī)劃分訓(xùn)練集和測(cè)試集X_train,X_test,y_train,y_test=train_test_split(X,Y,test_size=0.3,random_state=6)創(chuàng)建模型model=DecisionTreeClassifier(random_state=5).fit(X_train,y_train)對(duì)測(cè)試集進(jìn)行預(yù)測(cè)pre=model.predict(X_test)并將結(jié)果與label進(jìn)行對(duì)比，計(jì)算準(zhǔn)確度print("準(zhǔn)確度為：{:.3f}".format(np.mean(pre==y_test)))print("準(zhǔn)確度為