完整的程序如下:
>from sklearn.feature_extraction import DictVectorizer
>import csv
>from sklearn import tree
>from sklearn import preprocessing
>from sklearn.externals.six import StringIO
># 第一步:读取CSV文件
>def trainDicisionTree(csvfileurl):
> featureList = [] # 存储特征值
> labelList = [] # 存储预测的目标值
> allElectronicsData=open(csvfileurl)