决策树将标签和影响因子01化

决策树将标签和影响因子01化（数据01化）

import graphviz
from sklearn.feature_extraction import  DictVectorizer
from sklearn import tree
from sklearn import preprocessing
import csv
DTree=open(r'AllElectronics.csv','r')
reader=csv.reader(DTree)
headers=reader.__next__()
featureList=[]
labelList=[]
for row in reader:
    labelList.append(row[-1])
    rowDict={}
    for i in range(1,len(row)-1):
        rowDict[headers[i]]=row[i]
    featureList.append(rowDict)

vec=DictVectorizer()
x_data=vec.fit_transform(featureList).toarray()
lb=preprocessing.LabelBinarizer()
y_data=lb.fit_transform(labelList)

# print(vec.get_feature_names())
# print(x_data)
# print(y_data)