决策树算法优化(三)

在数据量很小的情况下,使用决策树能够很快的对数据进行分类,如果数据量大的话,构造决策树将耗费大量的时间,如果用创建好的决策树来进行分类能很好的解决这个问题。
使用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')

你可能感兴趣的:(机器学习)