在数据量很小的情况下,使用决策树能够很快的对数据进行分类,如果数据量大的话,构造决策树将耗费大量的时间,如果用创建好的决策树来进行分类能很好的解决这个问题。
使用pickle存储决策树,使用的时候直接读取。
def storeTree(inputTree,filename):
fw=open(filename,'wb')
pickle.dump(inputTree,fw)
fw.close()
def grabTree(filename):
fr=open(filename,'rb')
data=pickle.load(fr)
fr.close()
return data
storeTree(mytree,'mytree.pkl')
grabTree('mytree.pkl')